python - 构成 pandas dataframe 单元的数组的最大值

Question

我有一个 pandas 数据框，其中一列由数组组成。所以每个单元格都是一个数组。

假设数据框 df 中有一个列 A，这样

A = [ [1, 2, 3], 
      [4, 5, 6],
      [7, 8, 9], 
      ... ]

我想在每个数组中进行操作并获取，例如每个数组的最大值，并将其存储在另一列中。

在示例中，我想获取另一列

B = [ 3, 
      6,  
      9, 
      ...]

到目前为止，我已经尝试过这些方法，但没有一个给出我想要的。

df['B'] = np.max(df['A']);#

df.applymap (lambda B: A.max())

df['B'] = df.applymap (lambda B: np.max(np.array(df['A'].tolist()),0))

我应该如何进行？这是组织数据框的最佳方式吗？

Answer 1

您只需apply(max)即可。值是列表还是 np.array 并不重要。

df = pd.DataFrame({'a': [[1, 2, 3], [4, 5, 6], [7, 8, 9]]})
df['b'] = df['a'].apply(max)
print(df)

输出

           a  b
0  [1, 2, 3]  3
1  [4, 5, 6]  6
2  [7, 8, 9]  9