计算DataFrame中列表的平均值，忽略空的列表

Question

我有一个如下所示的数据框：

               A                    
    1  [67.0, 51.0, 23.0, 49.0, 3.0]    
    2  0
    3  [595.0]
    4  0
    5  [446.0, 564.0, 402.0]
    6  0 
    7  0

我想找到忽略零的每个列表的平均值。我希望得到类似的东西：

               A                     Mean
1  [67.0, 51.0, 23.0, 49.0, 3.0]     38.6
2  0                                    0
3  [595.0]                          595.0
4  0                                    0
5  [446.0, 564.0, 402.0]            470.7
6  0                                    0 
7  0                                    0

我尝试了很多可能的解决方案，但这些解决方案都没有。这是我到目前为止所尝试的：

df['Mean'] = df.A.apply(lambda x: mean(x)) 

这给了我这个错误

TypeError：'int'对象不可迭代

还有这个

df['Mean'] = df['A'].mean(axis=1)

ValueError：对象类型没有名为1的轴

试过这些也没有运气：

a = np.array( df['A'].tolist())
a.mean(axis=1)

mean(d for d in a if d)

还有什么我能试试的会给我预期的结果吗？谢谢你的帮助。

Answer 1

from collections.abc import Iterable
import numpy as np

def calculate_mean(x):
    if isinstance(x["A"], Iterable):
        x["mean"] = np.mean(x["A"])
    else:
        x["mean"] = x["A"]
    return x

df = df.apply(lambda x: calculate_mean(x), axis=1)

编辑 -

df["mean"] = df.apply(lambda x: np.mean(x["A"]), axis=1)

Answer 2

好吧这对我有用

                A                    
1   [67.0, 51.0, 23.0, 49.0, 3.0]    
2                               0
3                         [595.0]
4                               0
5           [446.0, 564.0, 402.0]
6                               0 
7                               0

使用np.mean

data['A'].apply(lambda x: np.mean(eval(x)))

产量

                A                            Mean
1   [67.0, 51.0, 23.0, 49.0, 3.0]       38.600000
2                               0       0.000000
3                         [595.0]       595.000000
4                               0       0.000000
5           [446.0, 564.0, 402.0]       470.666667
6                               0       0.000000
7                               0       0.000000

Answer 3

一种方法是使用列表推导并计算mean，其中给定行是列表，可以使用isinstance检查。这是必要的，否则你会得到：

TypeError：'int'对象不可迭代

因为函数期望可迭代。所以你可以这样做：

from statistics import mean
df['mean'] = [mean(i) if isinstance(i, list) else i for i in df.A]

              A                      mean
0  [67.0, 51.0, 23.0, 49.0, 3.0]   38.600000
1                              0    0.000000
2                        [595.0]  595.000000
3                              0    0.000000
4          [446.0, 564.0, 402.0]  470.666667
5                              0    0.000000
6                              0    0.000000

---

或者你也可以使用np.mean来处理ints和iterables：

import numpy as np
df['mean'] = df.A.map(np.mean)

               A                      mean
0  [67.0, 51.0, 23.0, 49.0, 3.0]   38.600000
1                              0    0.000000
2                        [595.0]  595.000000
3                              0    0.000000
4          [446.0, 564.0, 402.0]  470.666667
5                              0    0.000000
6                              0    0.000000