python - Pandas : extracting column1's value at column2th position

Question

我有一个数据框，其中有两列，一列是字符串列，一列是整数列。

column1  column2
 abcdef        2
  gtihj        4
  jiuwq        3

我想要另一个column3，其column1的值位于col2位置，如下所示

column1    column2   column3    
abcdef     2         b
gtihj      4         h
jiuwq      3         u

我尝试了以下代码

result['column1'].str.get(result['column2'])

但出现以下错误

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

Answer 1

生成错误是因为 str.get 需要标量索引，而不是索引列表/可迭代(它不理解这种输入，也不知道如何处理它) 。

在这种情况下，列表理解可以解决问题 -

df['column3'] = [i[j - 1] for i, j in zip(df.column1, df.column2)]

df    
  column1  column2 column3
0  abcdef        2       b
1   gtihj        4       h
2   jiuwq        3       u

这个问题从根本上来说很难向量化，因此以 C 速度运行的列表理解是一个非常高性能的替代方案。

较慢的替代方案包括(为了完整性)，apply，

df['column3'] = df.apply(lambda x: x.column1[x.column2 - 1], 1)

df    
  column1  column2 column3
0  abcdef        2       b
1   gtihj        4       h
2   jiuwq        3       u

和np.vectorize:

f = np.vectorize(lambda x, y: x[y - 1])
df['column3'] = f(df.column1, df.column2)

df
  column1  column2 column3
0  abcdef        2       b
1   gtihj        4       h
2   jiuwq        3       u