python - 为 Pandas 中的每一列获取非零值

Question

我有 Pandas 数据框作为df:

accel access adviser afpif  afp   publish  afraid verizon
0.00  0.14    0.00   0.00   0.00   0.13    0.00   0.44
0.13  0.00    0.00   0.77   0.00   0.00    0.22   0.00
0.00  0.00    0.87   0.00   0.34   0.00    0.00   0.00
......................................................
.....................................................

我还有一个列表L，其中包含作为元素的列名

L=['accel','afp','publish']

我只想根据pandas dataframe提取这些列表元素的非零值。

预期输出:-

dictionary={'accel':0.13,'afp':0.34,'publish':0.13}

Answer 1

使用DataFrame.loc听写理解和iat如果总是存在至少一个非 0 值:

d = {c: df.loc[df[c] ! =0, c].iat[0] for c in L }
print (d)
{'accel': 0.13, 'afp': 0.34, 'publish': 0.13}

更一般的也只使用 0 列:

d = {c: next(iter(df.loc[df[c] != 0, c]), 'no value') for c in L }
print (d)
{'accel': 0.13, 'afp': 0.34, 'publish': 0.13}