python：数据框列迭代在第一轮后停止

Question

我正在尝试打印数据帧的特定列中包含 1 的索引。 for 循环在第一次迭代时满足条件并返回它而不是向前移动。 我找遍了，发现其他人也有同样的问题，但我还是想不出解决办法。

import pandas as pd
d = {'a': [0.1, 0.2,0.3,0.4,0.5,0.6], 'b': [0.6, 0.8,0.3,0.4,0.1,0.1],
     'c': [0.7, 0.3,0.9,0.4,1.0,0.2],'d': [1,0,0,1,0,1]}
df = pd.DataFrame(data=d)
klist=[]
for i in (df.d):
    if  i == 1:
        klist.append(df.d.tolist().index(i))
           
print(klist) # should print [0, 3, 5] instead of [0, 0, 0]

Answer 1

使用索引而不是循环

klist = df[df['d'] == 1].index.tolist()
print(klist)

# Output:
[0, 3, 5]