Python数组操作将一维数组变成二维数组

一、问题

我们在进行数组操作的时候会遇到将一个低维的数组变成一个高维的素数组

二、解决

第一种方法
基本思路就是将低维数组进行等长的循环，在第一次为零的情况下，需要添加一个[]数组，原因是将它的基本框架搭建起来

1 records = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
2 result = []
3 for y in range(0, 4):
4     for x in range(0, 3):
5         if x == 0:
6             result.append([])
7         result[y].append(records[x + y * 3])
8 print(result)

第二种方法，numpy

2.1 升维度

利用函数reshape或者是resize

使用reshape的时候需要注意reshape的结果不改变，因此适用于还要用到原数组的情况

使用resize会改变原数组，因此适用于一定需要修改后的结果为值的情况

import numpy as np

x = np.arange(20)  # 生成数组
print(x)
result = x.reshape((4, 5))  # 将一维数组变成4行5列  原数组不会被修改或者覆盖
x.resize((2, 10))  # 覆盖原来的数据将新的结果给原来的数组
print(x)

2.1.1 结果

Python数组操作将一维数组变成二维数组

2.2 降维度

import numpy as np

arr = np.arange(10)
arr.resize((2, 5))
print(arr)
print(f"维度交换：\n{arr.swapaxes(1, 0)}")
print(f"C{arr.flatten('C')}")  # 默认C  一行为主
print(f"\nF:{arr.flatten('F')}")  # 以列为主
print(f"\nA:{arr.flatten('A')}")  # 和行一样
print(f"\nK:{arr.flatten('K')}")  # 和行一样

2.2.1 结果

Python数组操作将一维数组变成二维数组