python - 如何使用 Pandas 构建三个输入或门

Question

我的数据帧 df 有 3 个输入（A、B、C），如下所示

A   B   C
0   0   0
0   0   1
0   1   0
0   1   1
1   0   0
1   0   1
1   1   0
1   1   1

我想要构建逻辑或门并具有如下所示的示例输出

A   B   C   Output
0   0   0   0
0   0   1   1
0   1   0   1
0   1   1   1
1   0   0   1
1   0   1   1
1   1   0   1
1   1   1   1

如何在熊猫中做到这一点

标签： pythonpython-3.xpandasnumpylogical-or

Answer 1

您只需要评估df.A | df.B | df.C.

df['OR_Gate'] = df.A | df.B | df.C

注意：如果 A、B、C 列中的值是 0 和 1 的字符串，则执行以下操作之一：

# Method-1: 
#   Convert the strings into int and then evaluate OR_Gate: 
#   This changes the value-types in the columns A, B, C
df = df.astype('int')
df['OR_Gate'] = df.A | df.B | df.C
# Method-2: 
#   This will not change the original data type in columns A, B, C
#   But will correctly evaluate 'OR_Gate'.
df['OR_Gate'] = df.A.astype(int) | df.B.astype(int) | df.C.astype(int)
# Method-3: 
#   If you want your final output to be in boolean form.
df['OR_Gate'] = df.A.astype(bool) | df.B.astype(bool) | df.C.astype(bool)

详细解决方案

import pandas as pd

# Dummy data
A = [0]*4 + [1]*4
B = [0]*2 + [1]*2 + [0]*2 + [1]*2
C = [0, 1]*4
# Make Dataframe
df = pd.DataFrame({'A': A, 'B': B, 'C': C})
# Update 'OR_Gate' Output
df['OR_Gate'] = df.A | df.B | df.C
df

输出：

   A  B  C  OR_Gate
0  0  0  0        0
1  0  0  1        1
2  0  1  0        1
3  0  1  1        1
4  1  0  0        1
5  1  0  1        1
6  1  1  0        1
7  1  1  1        1