파이썬 넘파이

파이썬 넘파이!!

자주 쓰이는 걸 찾아서

모양 변경(reshape)

• 3D array to 2D

a = np.array(
[[[8, 4, 1, 0, 0],
[6, 8, 5, 5, 2],
[1, 1, 1, 1, 1]],

[[8, 7, 1, 0, 3],
[2, 8, 5, 5, 2],
[1, 1, 1, 1, 1]],

[[2, 4, 0, 2, 3],
[2, 5, 5, 3, 2],
[1, 1, 1, 1, 1]]])

a.transpose([1,0,2]).reshape(3,15)
>>
array([[8, 4, 1, 0, 0, 8, 7, 1, 0, 3, 2, 4, 0, 2, 3],
[6, 8, 5, 5, 2, 2, 8, 5, 5, 2, 2, 5, 5, 3, 2],
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]])

•  2D to 3D array

data = [[11, 22],
[33, 44],
[55, 66]]

data = array(data)
print(data.shape)
>>(3,2)
data = data.reshape((data.shape[0], data.shape[1], 1))
print(data.shape)
>>(3,2,1)

필요한 부분만큼 가져오기 (slice)

공식은 : data[from: to]

• 1차원 자르기

data = array([11, 22, 33, 44, 55])
print(data[:])
>>[11, 22, 33, 44, 55]

print(data[0:1])
>>[11]

print(data[-2:])
>>[44,55]

• 2차원 자르기

data = array([[11, 22, 33],
		[44, 55, 66],
		[77, 88, 99]])
X, y = data[:, :-1], data[:, -1]
print(X)
>>[[11 22]
 [44 55]
 [77 88]]

print(y)
>>[33 66 99]


data = array([[11, 22, 33],
		[44, 55, 66],
		[77, 88, 99]])

split = 2
train,test = data[:split,:],data[split:,:]
print(train)
>> [[11 22 33]
[44 55 66]]

print(test)
>> [[77 88 99]]