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

【问题标题】：How can I use values from a Pandas data frame in a calcul?如何在计算中使用 Pandas 数据框中的值？
【发布时间】：2022-07-17 04:11:00
【问题描述】：

我有一个 Pandas 数据框。在A 列中有类似[1, 5, 3] 的整数，在B 列中有类似["abcdef", "ghijklmno", "qwertyuiop"] 的字符串

我想根据A 列创建一个C 列，其中B 列的第一个字符。在我的示例中，我希望 C 列类似于 ["a", "ghijk", "qwe" ]

我试过了：

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

但它不起作用。

【问题讨论】：

标签： python pandas string dataframe

【解决方案1】：

您可以设置使用lambda函数并设置axis = 1以使用A列设置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

【讨论】：

【解决方案2】：

也许尝试遍历列：

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

【讨论】：