使用:
np.random.seed(12)
df = pd.DataFrame(np.random.randint(3,size=(10,2)), columns=['Sex','Pclass'])
df['prediction'] = ((df['Sex'] == 1) & (df['Pclass'] == 1)).astype(int)
print (df)
Sex Pclass prediction
0 2 1 0
1 1 2 0
2 0 0 0
3 2 1 0
4 0 1 0
5 1 1 1
6 2 2 0
7 2 0 0
8 1 0 0
9 0 1 0
すべての値は、1
であり、0
のみJohn Galt溶液を使用する場合:
#only 0, 1 values
df['predictions'] = df.all(axis=1).astype(int)
#if more possible values
df['predictions'] = df.eq(1).all(axis=1).astype(int)
print (df)
Sex Pclass predictions
0 2 1 0
1 1 2 0
2 0 0 0
3 2 1 0
4 0 1 0
5 1 1 1
6 2 2 0
7 2 0 0
8 1 0 0
9 0 1 0
'df.all(1).astype(INT)'または '' df.eq(1).ALL(1).astype(INTを) "おそらく? – Zero
@JohnGalt、ありがとうございます!あなたのバージョンを答えに加えました... – MaxU