Numpy 차원축소

□ 차원추가 : .expand_dims(변수명, axis = 0 or 1)

□ 차원축소 : .squeeze(변수명, axis = 0 or 1)

#1차원
arr=np.array([1,2])
print(f"arr :\n{arr}")
print(f"arr 차원 :\n{arr.ndim}")
print("\n")

#차원 추가
arr1=np.expand_dims(arr,axis=1)
print(f"arr 차원추가 행방향 :\n{arr1}")
print("\n")
arr2=np.expand_dims(arr,axis=0)
print(f"arr 차원추가 열방향 :\n{arr2}")
print("\n")

print(f"arr1 :\n{arr1}")
print("\n")
print(f"arr2 :\n{arr2}")
print("\n")

#차원축소
arr3=np.squeeze(arr1, axis=1)
arr4=np.squeeze(arr2, axis=0)
print(f"arr1 차원축소 행방향 :\n{arr3}")
print("\n")
print(f"arr2 차원축소 열방향 :\n{arr4}")