我有一个名为
我希望找到具有非零行值的列的名称。每行中非零值的列名应存储在列表中。例如,以下是
预期结果:
df
的pandas数据框。df = pd.DataFrame(np.array([[0.6, 0.4, 0], [0.0, 1.0, 0], [0.4, 0.6, 0]]),columns = [-5, 0, 5])
print(df)
-5 0 5
0 0.6 0.4 0.0
1 0.0 1.0 0.0
2 0.4 0.6 0.0
我希望找到具有非零行值的列的名称。每行中非零值的列名应存储在列表中。例如,以下是
df
中三行的预期结果。预期结果:
[[-5,0],[0],[-5,0]]