我有一个2D数组,看起来像这样:
array = [[23 ,89, 4, 3, 0],[12, 73 ,3, 5,1],[7, 9 ,12, 11 ,0]]
对于所有行,最后一列始终为0或1.我的目标是为第0列计算两个均值,其中一个均值将是最后一列的值为0,其中一个均值将是最后一列的值为1.
例如,对于上面给定的样本数组:平均值1:15(对于最后一列为0的所有行,对于0列的平均值)意味着2:12(对于最后一列为1的所有行,均值为0列)
我试过这个(火车是我输入数组的名字):
mean_c1_0=np.mean(train[:: , 0])
variance_c1_0=np.var(train[:: , 0])
这让我得到了第0列ll值的均值和方差.
我总是可以引入一个for循环和几个if条件来继续检查最后一列,然后在第0列添加相应的值,但我正在寻找一种有效的方法.由于我是Python的新手,我希望是否有一个可以完成这项工作的numpy函数.
你能指点我任何这样的文件吗？