如何在计算中使用 Pandas 数据框中的值？

Question

I have a Pandas data frame. In the A columns there are ints like [1, 5, 3] , and in the B columns there are string like ["abcdef", "ghijklmno", "qwertyuiop"]

I want to create a C columns with the columns B first char according to the columns A . In my example I want the C columns to be like ["a", "ghijk", "qwe" ]

I tried:

data_frame['C'] = data_frame.B.str[:data_frame["A"]]

but it doesn't work.

Answer 1

You can set use a lambda function and set the axis = 1 to use column A to set the string length of B

df = pd.DataFrame({
    'A' : [1, 5, 3,],
    'B' : ["abcdef", "ghijklmno", "qwertyuiop"]
})
df['c'] = df.apply(lambda x : x['B'][:x['A']], axis = 1)
df

Answer 2

也许尝试遍历列：

df['C'] = [b[:a] for a, b in zip(df['A'], df['B'])]