Prepare > Python > Numpy > Mean, Var, and Std

Mean, Var, and Std | HackerRank

 

문제

---

Task

You are given a 2-D array of size X.
Your task is to find:

1. The mean along axis 1
2. The var along axis 0
3. The std along axis None

Input Format

The first line contains the space separated values of N and M.
The next N lines contains M space separated integers.

Output Format

First, print the mean.
Second, print the var.
Third, print the std.

 

=> N X M의 array를 받아서 (열)평균, (행)분산, 표준편차를 구하라

 

 

 

 

코드

---

import numpy as np

N, M = map(int, input().split())


arr = np.array(
    [input().split() for _ in range(N)], float
)

print(np.mean(arr, 1))
print(np.var(arr, 0))
print(round(np.std(arr), 11))

 

 

 

 

노트

---

np.mean()

import numpy

my_array = numpy.array([ [1, 2], [3, 4] ])

print(numpy.mean(my_array, axis = 0))        #Output : [ 2.  3.]
print(numpy.mean(my_array, axis = 1))        #Output : [ 1.5  3.5]
print(numpy.mean(my_array, axis = None))     #Output : 2.5
print(numpy.mean(my_array))                  #Output : 2.5

• axis=0: 행 방향(가로)에서 평균 리턴
• axis=1: 열 방향(세로)에서 평균 리턴
• Defualt는 None으로 array의 모든 요소의 평균 리턴

 

 

np.var()

import numpy

my_array = numpy.array([ [1, 2], [3, 4] ])

print(numpy.var(my_array, axis = 0))         #Output : [ 1.  1.]
print(numpy.var(my_array, axis = 1))         #Output : [ 0.25  0.25]
print(numpy.var(my_array, axis = None))      #Output : 1.25
print(numpy.var(my_array))                   #Output : 1.25

• axis=0: 행 방향(가로)에서 분산 리턴
• axis=1: 열 방향(세로)에서 분산 리턴
• Defualt는 None으로 array의 모든 요소의 분산 리턴

 

 

np.std

import numpy

my_array = numpy.array([ [1, 2], [3, 4] ])

print(numpy.std(my_array, axis = 0))         #Output : [ 1.  1.]
print(numpy.std(my_array, axis = 1))         #Output : [ 0.5  0.5]
print(numpy.std(my_array, axis = None))      #Output : 1.11803398875
print(numpy.std(my_array))                   #Output : 1.11803398875

• axis=0: 행 방향(가로)에서 표준편차 리턴
• axis=1: 열 방향(세로)에서 표준편차 리턴
• Defualt는 None으로 array의 모든 요소의 표준편차 리턴

 

 

round()

import numpy as np

N, M = map(int, input().split())

arr = np.array(
    [input().split() for _ in range(N)], float
)

print(arr_std)              # 1.118033988749895
print(round(arr_std), 11)   # 1.11803398875

print(round(-0.1))  # 0
print(round(-1.7))  # -2
print(round(-0.12, 1))  # -0.1
print(round(-1.76, 1))  # -1.8

print(round(6.5))  # 6
print(round(5.5))  # 6
print(round(4.5))  # 4 
print(round(3.5))  # 4
print(round(2.5))  # 2
print(round(1.5))  # 2
print(round(0.5))  # 0


print(round(-6.5))  # -6
print(round(-5.5))  # -6
print(round(-4.5))  # -4
print(round(-3.5))  # -4
print(round(-2.5))  # -2
print(round(-1.5))  # -2
print(round(-0.5))  # 0

• 첫 번째 파라미터(number= ): 반올림 할 대상 숫자
• 두 번재 파라미터(ndigits= ): 보고싶은 자릿수
• ndigits를 생략하면 number에 가장 가까운 정수를 반환(Defualt는 0인듯..?)
• 가장 가까운 정수가 두 개라면, 짝수를 리턴(음수인 경우에는 절댓값이 짝수인 수)

 

 

 

 

참조

---

[python] 파이썬 반올림 round 함수에 대해서