查找在最后一列上筛选的矩阵的所有列的平均值

2024-03-29 09:13:24 发布

您现在位置:Python中文网/ 问答频道 /正文

我是新人。我有一个Nx4矩阵,我想找到每列的平均值,比如最后一列等于1。在matlab中,我会做一些类似于mean1 = mean(data[column 4] == 1)的事情。这将返回一个矩阵(或向量)和列的平均值,列4的平均值等于1。我找不到任何具体的文档来说明如何处理这个问题。This显示了如何过滤矩阵,但我不必将矩阵重新分配给一个新变量,从而使存储大小加倍。提前谢谢。你知道吗


Tags: 文档datacolumn矩阵meanthis事情向量
1条回答
网友
1楼 · 发布于 2024-03-29 09:13:24
#make artificial data to match problem
data = np.random.random((100,4))
print( id(data) )
data[:,3] = data[:,3] < 0.5
print( id(data) )  #same object (memory location)
#get the filter
dfilter = data[:,3].astype(np.bool_)
#find the means
means = data[dfilter].mean(axis=0)

相关问题 更多 >