NumPy系统是Python的一种开源的数值计算扩展。这种工具可用来存储和处理大型矩阵,比Python自身的嵌套列表(nested list structure)结构要高效的多(该结构也可以用来表示矩阵(matrix))。(来自百度百科)
很多人以为Python的list就是数组,其实是错误的,Python的数组应该为numpy,而不是我们常用的list。那么,下面我们来操作下数组、列表、矩阵之间的相互转换。
import numpy as np
my_list = [[1, 2, 3], [4, 5, 6]] # 列表定义
print('列表:', end='')
print(type(my_list))
print(my_list, end='\n\n')
my_array = np.array([[1, 2, 3], [4, 5, 6]]) # 数组定义(这里是二维数组)
print('数组:', end='')
print(type(my_array))
print(my_array, end='\n\n')
my_matrix = np.mat([[1, 2, 3], [4, 5, 6]]) # 矩阵定义
print('矩阵:', end='')
print(type(my_matrix))
# print(my_matrix[0:1, 0:2]) # 取矩阵的第0行,第0到第1列的数据(含左不含右)
print(my_matrix, end='\n\n')
# 下面是列表、数组、矩阵之间的相互装换
print('-----------------------我是分割线----------------------')
print('下面是列表、数组、矩阵之间的相互装换:')
list_to_array = np.array(my_list) # 列表转数组
print('列表转数组:', end='')
print(type(list_to_array))
print(list_to_array, end='\n\n')
list_to_matrix = np.mat(my_list) # 列表转矩阵
print('列表转矩阵:', end='')
print(type(list_to_matrix))
print(list_to_matrix, end='\n\n')
print('-----------------------我是分割线----------------------')
array_to_list = my_array.tolist() # 数组转列表
print('数组转列表:', end='')
print(type(array_to_list))
print(array_to_list, end='\n\n')
array_to_matrix = np.mat(my_array) # 数组转矩阵
print('数组转矩阵:', end='')
print(type(array_to_matrix))
print(array_to_matrix, end='\n\n')
print('-----------------------我是分割线----------------------')
matrix_to_list = my_matrix.tolist() # 矩阵转列表
print('矩阵转列表:', end='')
print(type(matrix_to_list))
print(matrix_to_list, end='\n\n')
matrix_to_array = np.array(my_matrix) # 矩阵转数组
print('矩阵转数组:', end='')
print(type(matrix_to_array))
print(matrix_to_array, end='\n\n')
执行结果:
列表:<class 'list'>
[[1, 2, 3], [4, 5, 6]]
数组:<class 'numpy.ndarray'>
[[1 2 3]
[4 5 6]]
矩阵:<class 'numpy.matrixlib.defmatrix.matrix'>
[[1 2 3]
[4 5 6]]
-----------------------我是分割线----------------------
下面是列表、数组、矩阵之间的相互装换:
列表转数组:<class 'numpy.ndarray'>
[[1 2 3]
[4 5 6]]
列表转矩阵:<class 'numpy.matrixlib.defmatrix.matrix'>
[[1 2 3]
[4 5 6]]
-----------------------我是分割线----------------------
数组转列表:<class 'list'>
[[1, 2, 3], [4, 5, 6]]
数组转矩阵:<class 'numpy.matrixlib.defmatrix.matrix'>
[[1 2 3]
[4 5 6]]
-----------------------我是分割线----------------------
矩阵转列表:<class 'list'>
[[1, 2, 3], [4, 5, 6]]
矩阵转数组:<class 'numpy.ndarray'>
[[1 2 3]
[4 5 6]]