python - 基于列名称扩展的DataFrame列的平均值

我在Jupiter中有一个DataFrame A，看起来像下面的样子

Index Var1.A.1  Var1.B.1  Var1.CA.1  Var2.A.1  Var2.B.1  Var2.CA.1
0      1         21         3          3         4        4
1      3         5          4          9         5        1
....
100    9         75        2           4         8        2

我想根据名称的扩展名来评估平均值，即


.A.1的平均值
.B.1的平均值
.CA.1的平均值


例如，为了评估扩展名为.A.1的变量的平均值，我尝试了以下操作，但未返回我想要的结果

List=['.A.1', '.B.1', '.CA.1']
A[List[List.str.contains('.A.1')]].mean()

但是，通过这种方式，我得到了不同变量的平均值，也得到了CA.1，这不是它要查找的内容。

有什么建议吗？

谢谢

最佳答案

如果要在第一个mean之后按所有值按每行.使用groupby和lambda函数以及mean：

df = df.groupby(lambda x: x.split('.', 1)[-1], axis=1).mean()
print (df)
     A.1   B.1  CA.1
0    2.0  12.5   3.5
1    6.0   5.0   2.5
100  6.5  41.5   2.0

关于python - 基于列名称扩展的DataFrame列的平均值，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/55298387/