python - 在 pandas 列中插入字典值

Question

C 是我在 pandas dataframe(df) 中的专栏。它由许多列表组成。

          C
  [ab ab bc abb]
  [ll li lo ll]

D是我的字典，如下。

D={'ab':0, 'bc':1, 'abb':2, 'll':3, 'li':4, 'lo':5}

现在，我想将字典的值分配给列表，为此，我使用下面提到的代码。

df.C= [D[item] for item in df.C]

在这里，我收到此错误:

类型错误:列表索引必须是整数或切片，而不是字符串。

感谢您的帮助。

Answer 1

看起来您有一个空格分隔字符串列表，因此您需要两个循环，而不是一个。试试这个:

df.C = [[D[j] for j in i.split()] for i df.C]

如果您必须处理丢失的键，请使用 dict.get 代替:

df.C = [[D.get(j, -1) for j in i.split()] for i df.C]

其中-1是一些默认值。