I have the following code

df = pd.DataFrame(columns=[i for i in range(52)], index=[j for j in range(len(all_sentence_id_info))])

I also have a list of particular indices called sentence_embedding_replacement_id, I want color just the index cells, and not the entire row, with a color if the number in that index cell is found in sentence_embedding_replacement_id.

I tried using df.style.apply but for some reason I never got my helper function to get called, I tried using lambda but I kept getting the error 'Series' object has no attribute 'columns'