循环遍历熊猫中的行[重复]

Question

我正在使用 python，在 python 中循环遍历行时遇到了一些困难。 我的数据框包含 3 列：id、val1、大小。 我想根据大小创建列 col1。 我尝试这段代码，但我的代码永远不在第一个条件之内。请问我应该如何改正。假设我不会与其他方法一起工作，我只是想解决我自己的代码。

友好，

数据示例

data = [['x1', 100, 1], ['x2', 200, 2], ['x3', 300, 1]]
df = pd.DataFrame(data, columns = ['id', 'val1', 'size'])

代码

if (df['size'] == 2) is True:
  df['col1'] = df['val1']
  print("1")
else:
  pass

Answer 1

你想要print吗？您可以在没有循环的情况下执行此操作，如下所示：

df['col1'] = np.where(df['size'] == 2, df['val1'], np.nan)

输出：

>>> df
    id  val1    size    col1
0   x1  100     1   NaN
1   x2  200     2   200.0
2   x3  300     1   NaN

Answer 2

df['col1'] = df.loc[df['size'] == 2, 'val1']
print(df)

   id  val1  size   col1
0  x1   100     1    NaN
1  x2   200     2  200.0
2  x3   300     1    NaN

Answer 3

如果你像这样编辑你的代码，你可以制作 col1：

data = [['x1', 100, 1], ['x2', 200, 2], ['x3', 300, 1]]
df = pd.DataFrame(data, columns=['id', 'val1', 'size'])

col1 = []
for i, row in df.iterrows():
    if row['size'] == 2:
        col1.append(row['val1'])
    else:
        col1.append(None)
df['col1'] = pd.Series(col1)

print(df)

另一种方式是：

def func(df):
    if df['size'] == 2:
        return df['val1']
    else:
        return None

df['col1'] = df.apply(func, axis=1)
print(df)

这将打印：

   id  val1  size   col1
0  x1   100     1    NaN
1  x2   200     2  200.0
2  x3   300     1    NaN