均值,变量和标准

问题描述

任务是 您将得到一个大小为X的二维数组。 您的任务是找到:

  1. 沿轴的平均值
  2. 沿轴的var
  3. 沿轴的std

输入格式 第一行包含和的空格分隔值。 接下来的行包含用空格分隔的整数。

输出格式 首先,打印均值。 第二,打印var。 第三,打印标准。

样本输入

2 2
1 2
3 4

示例输出

[ 1.5  3.5]
[ 1.  1.]
1.11803398875

我的代码

import numpy
N,M = map(int,input().split(" "))
A = numpy.array([input().split() for _ in range(N)],int)
print(numpy.mean(A,axis = 1))
print(numpy.var(A,axis = 0))
print(round(numpy.std(A,axis = None),11))

输出

enter image description here

我似乎有一些缩进问题或打印结果超出预期,该数组的第一个垂直元素前面有一个空格。 我做错什么了吗?

解决方法

在导入numpy和打印numpy数据之间的任何位置。它告诉numpy打印格式化程序使用numpy版本1.13中的默认设置,而不是numpy版本1.14(当前版本)中的默认设置。问题设定的结果是固定的,并且显然是用旧的numpy完成的,因此,如果不这样做,则会出现各种格式不匹配的情况,即使您的实际答案正确,也会导致测试失败。

因此请使用np.set_printoptions(legacy='1.13')

import numpy as np 

n,m = map(int,input().split())
b = []
for i in range(n):
    a = list(map(int,input().split()))
    b.append(a)

b = np.array(b)

np.set_printoptions(legacy='1.13')
print(np.mean(b,axis = 1))
print(np.var(b,axis = 0))
print(np.std(b))