python - 数据帧列表理解 "zip(...)": loop through chosen df columns efficiently with just a list of column name strings

Question

这只是一个挑剔的语法问题......

我有一个数据框，我想使用列表理解来评估使用大量列的函数。

我知道我能做到

df['result_col'] = [some_func(*var) for var in zip(df['col_1'], df['col_2'],... ,df['col_n'])]

我想做这样的事情

df['result_col'] = [some_func(*var) for var in zip(df[['col_1', 'col_2',... ,'col_n']])]

即不必编写 df n 次。我一生都无法弄清楚语法。

Answer 1

这应该可行，但老实说，OP自己也想到了，所以+1 OP:)

df['result_col'] = [some_func(*var) for var in zip(*[df[col] for col in ['col_1', 'col_2',... ,'col_n']])]