python - Pandas Dataframe - 在特定行中选择具有特定值的列

Question

我想在 Pandas Dataframe 的特定行（比如第一行）中选择具有特定值（比如 1）的列

标签： pythonpandasdataframe

Answer 1

您可以使用iloc将一行提取为一个系列，然后应用您的条件：

row = df.iloc[0]           # extract first row as series
res = row[res == 1].index  # filter for values equal to 1 and get columns via index