python - 根据条件从现有数据框创建新数据框

Question

我有一个数据框 df

import pandas as pd
import numpy as np
df = pd.DataFrame(np.array([[0,1,1,0,1,0], [1,0,1,1,0,0], [1,1,0,0,0,1],[1,0,1,0,1,1], 
[0,0,1,0,0,1]]))
df

现在，从数据框 df 我喜欢根据条件创建一个新的数据框条件:如果某列包含三个或三个以上“1”，则新数据框列值为“1”，否则为“0”

expected output of new data frame
    1 0 1 0 0 1

Answer 1

您也可以在不apply的情况下获得它。您可以沿行sum，axis=0，并使用gt(2)创建一个 bool 值:

res = df.sum(axis=0).gt(2).astype(int)

print(res)

0    1
1    0
2    1
3    0
4    0
5    1
dtype: int32

正如 David 指出的，上面的结果是一个系列。如果您需要数据框，可以链接 to_frame()在它的最后