python - 迭代数据并将其输入到 pandas 数据框中的列中

Question

我有一个 pandas 数据框，其中有一列是一小部分字符串。我们将该列称为“A”，其中的所有值都是 string_1 , string_2 , string_3 .

现在，我想添加另一列并用与字符串对应的数值填充它。

我创建了一本字典

d = { 'string_1' : 1, 'string_2' : 2, 'string_3': 3}

然后我初始化了新列:

df['B'] = pd.Series(index=df.index)

现在，我想用整数值填充它。我可以通过以下方式调用与字典中的字符串关联的值:

for s in df['A']:
   n = d[s]

效果很好，但我尝试过使用简单的 df['B'] = n填充 for 循环中的新列，但这不起作用，我尝试用 pandas 找出索引。

Answer 1

如果我理解正确的话，你可以直接调用map:

df['B'] = df['A'].map(d)

这将执行查找并填充您要查找的值。