当前位置: 代码迷 >> 综合 >> numpy ndarray 形状(shape)变换(reshape)变形
  详细解决方案

numpy ndarray 形状(shape)变换(reshape)变形

热度:90   发布时间:2023-12-15 17:43:23.0

1,新建array (numpy.ndarray)

import numpy as np# 手动填写一个
a=[[1,2],[2,3],[3,4]]
a = np.array(a)# 随机生成一个
b = np.random.randint(0,10,(2, 3)) # 两行三列,元素从0到10

2,查看形状

print(a.shape)
# (3, 2)

3,多种变形

# 填写元素个数,变成一维
a.reshape(6)# 只给行数n,numpy根据给出的行数,自行计算(列参数为-1,注意元素总的个数要能被n整除)
# 或者是在不知道转换之后a的列数应该是多少的情况下使用。
a.reshape(2, -1)
# array([[1, 2, 2],
#       [3, 3, 4]])a.reshape(3, -1)
# array([[1, 2],
#       [2, 3],
#       [3, 4]])# 只给列数。同理。在不知道a的行数的情况下也可以使用。
a.reshape(-1, 1)
# array([[1],
#        [2],
#        [2],
#        [3],
#        [3],
#        [4]])