numpy的基本使用

import numpy as np

#创建一维数组arr1
arr1 = np.arange(1, 5, 1)
print(arr1)
print(arr1.shape)

#创建一维数组arr2
arr2 = np.array([1, 1, 1, 1])
print(arr2)

#数组相加
arr3 = arr1 + arr2
print(arr3)

#更改数组维度
arr1.shape = 2,2 #arr1.reshape(2,2)
print(arr1)

输出

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