如何删除pandas中仅包含零的列？

Question

我目前有一个数据框，其中包含以1和0作为值的列，我想遍历这些列并删除仅由0组成的列。到目前为止，这是我尝试过的：

ones = []
zeros = []
for year in years:
    for i in range(0,599):
        if year[str(i)].values.any() == 1:
            ones.append(i)
        if year[str(i)].values.all() == 0:
            zeros.append(i)
    for j in ones:
        if j in zeros:
            zeros.remove(j)
    for q in zeros:
        del year[str(q)]

在哪几年是我正在分析的不同年份的数据帧的列表，其中包括其中包含一个的列，而零则是包含所有零的列的列表。是否有更好的方法根据条件删除列？由于某些原因，我必须检查一列是否也位于零列表中，并将它们从零列表中删除，以获得所有零列的列表。

Answer 1

df.loc[:, (df != 0).any(axis=0)]

以下是其工作方式的细分：

In [74]: import pandas as pd

In [75]: df = pd.DataFrame([[1,0,0,0], [0,0,1,0]])

In [76]: df
Out[76]: 
   0  1  2  3
0  1  0  0  0
1  0  0  1  0

[2 rows x 4 columns]

df != 0创建一个布尔数据框，该框为True，其中df非零：

In [77]: df != 0
Out[77]: 
       0      1      2      3
0   True  False  False  False
1  False  False   True  False

[2 rows x 4 columns]

(df != 0).any(axis=0)返回一个布尔系列，指示哪些列具有非零条目。（该any操作将沿0轴（即沿行）的值聚合为一个布尔值。因此，结果是每列一个布尔值。）

In [78]: (df != 0).any(axis=0)
Out[78]: 
0     True
1    False
2     True
3    False
dtype: bool

而且df.loc可以用来选择那些列：

In [79]: df.loc[:, (df != 0).any(axis=0)]
Out[79]: 
   0  2
0  1  0
1  0  1

[2 rows x 2 columns]

要“删除”零列，请重新分配df：

df = df.loc[:, (df != 0).any(axis=0)]