在熊猫列中插入字典值

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]

在这里，我收到此错误：

TypeError: 列表索引必须是整数或切片，而不是 str。

感谢您的帮助。

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 是一些默认值。