在 Python 中,是否可以编写一个返回多维数组维度的函数(假设数组的维度不是锯齿状的)?
例如, 的尺寸为[[2,3], [4,2], [3,2]]
,[3, 2]
而 的尺寸[[[3,2], [4,5]],[[3,4],[2,3]]]
为[2,2,2]
。
Python 是否有任何内置函数可以返回多维数组的所有维度,还是我需要自己实现这个函数?
在 Python 中,是否可以编写一个返回多维数组维度的函数(假设数组的维度不是锯齿状的)?
例如, 的尺寸为[[2,3], [4,2], [3,2]]
,[3, 2]
而 的尺寸[[[3,2], [4,5]],[[3,4],[2,3]]]
为[2,2,2]
。
Python 是否有任何内置函数可以返回多维数组的所有维度,还是我需要自己实现这个函数?
不,没有任何内置的,因为这样的“数组” 1它可以是锯齿状的,“尺寸”或“形状”的概念根本没有任何意义。所以,你必须自己写。如果您可以假设所有维度上的一致性,您可以按照以下步骤进行:
dim1 = len(a)
dim2 = len(a[0])
dim3 = len(a[0][0])
.
.
.
让这个递归来处理所有维度是很容易的。这应该这样做:
def dim(a):
if not type(a) == list:
return []
return [len(a)] + dim(a[0])
但是,如果您需要这样的东西,您可能需要考虑查看NumPy数组,numpy.ndarray.shape
它可以为您提供所需的内容。
from numpy import array
l = [[2, 3], [4, 2], [3, 2]]
a = array(l)
print a.shape
输出
(3, 2)
1在吓人的引号中,因为您不是真的在看数组,而是在看一个列表,或列表列表,或列表列表...。
你可以用 numpy 做到这一点:
import numpy
l = [[2,3], [4,2], [3,2]]
m = numpy.array(l)
print m.shape
但是你的第二个例子的形状是[2,2,2],而不是[1,4,5],除非我误解了你的问题......
那不是一个多维数组。它是一个list
. 它恰好包含其他列表。没有什么可以说您的列表不能是:
[[2,3], [4,2], [3,2,4,5,6]]
在这种情况下,您希望这样的函数返回什么值?
没有通用函数可以满足您的要求,尤其是因为 Python 本身没有定义矩阵/数组类。如果您准备对列表的一致性做出假设或编写断言,您当然可以编写自己的函数,该函数对列表和元组等可迭代对象进行操作。用于len(a)
第一个维度,len(a[0])
用于第二个维度,依此类推。递归将成为您的朋友。
如果您对矩阵使用 numpy 数组,老实说这很有意义,那么您的函数将存在(它是shape
ndarray 类的属性)并且是有意义的。
假设输入数组不是锯齿状的:
def arr_dimen(a):
return [len(a)]+arr_dimen(a[0]) if(type(a) == list) else []
import numpy as np
# define array 3 row and 5 colums
a = np.arange(15).reshape(3, 5)
print ("the array elements:\n",a);
# print all dim
print("all dim :\n",a.shape)
# the print the number of row only
print("the number of rows :\n",a.shape[0])
# the print the number of colume only
print("the number of cols :\n",a.shape[1])