根据python中另一列的值提取字符串的一部分

Question

我想在另一列（A）的值减去1的基础上，增加列（B）的第一部分以获得输出。例如，如果A列是4，那么我需要B列的前3个字符

有什么建议吗？非常感谢

+---+---------------+------+
| A |       B       | Out  |
+---+---------------+------+
| 4 |  CTAAT        |  CTA |
| 3 |  GKAEI        |  GK  |
+---+---------------+------+

即第1行B列的前3个字母，第2行B列的前2个字母

我目前的方法不起作用：

df['output']= df.B.str[0:(df[A]-1)]

Answer 1

您可以使用pandas.apply：

df['output']=df.apply(lambda x: x['B'][:x['A']-1], axis=1)
print(df)

A      B         output
4      CTAAT     CTA
3      GKAEI     GK