首页 > itarticle > numpy basic 2

numpy basic 2

admin 11月 13, 2020 0

import numpy as np

A = np.arange(0, 10)
B = np.array([[1, 2], [2, 3], [3, 4]])


print(A)
print(B)
print(B[np.array([0,2])])
print(B[np.array([0,2])][np.array([1])])

res:
[0 1 2 3 4 5 6 7 8 9]
[[1 2]
 [2 3]
 [3 4]]
[[1 2]
 [3 4]]
[[3 4]]

2

import numpy as np

A = np.array([1, 2, 3, 4])
B = np.array([[1, 2, 3, 4],
              [2, 4, 5, 7]])

A_ = A.reshape(1, A.size)
print(A_)
print(A_.ndim)
print(A.ndim)
print(B.ndim)

res:
[[1 2 3 4]]
2
1
2