Python Pandas：比较两列和返回字符串值，如果第一列包含字符中的字符

Question

我首先有2列，其中包括X或F在内的字符串，第二列为空。 如果第1列中有任何X 每次我运行代码时，都将值分配给所有这些

这个示例应该如何：

我的代码：

for row in df['Column2']:
    if df['Column1'].str.contains('X').any():
        df['Column2'] = 'YES'
    else:
        df['Column2'] = 'NO'

Answer 1

您每次通过循环执行矢量化操作。每次通过循环，您都将'是''分配给整个column2。

使用numpy您可以做：

import numpy as np

df['Column2'] = np.where(df['Column1'].str.contains('X'), 'YES', 'NO')
print(df)

结果

             Column1 Column2
0  ....X.X.X.X..X.X.     YES
1  ....X.X.X.X..X.X.     YES
2  ....X.X.X.X..X.X.     YES
3  ....X.X.X.X..X.X.     YES
4          ....F.F.F‬      NO
5          ....F.F.F      NO
6          ....F.F.F      NO
7          ....F.F.F      NO

Answer 2

您可以使用Regex查找“ X”，

if df['Column1'].str.find(r'X')>1:

甚至可以避免循环如下

(df['Column1'].str.find(r'X')>1).map({True: 'Yes', False: 'No'})