在 Python 中获取二维数组的长度

Python 2年前 555

在 Python 中获取二维数组的长度：

将整个数组传递给 len() 函数以获取行数。
将第一个数组元素传递给 len() 函数以获取列数。
将行数乘以列数得到总数。

my_arr = [[1, 2, 3], [4, 5, 6]]

rows = len(my_arr)
print(rows)  # ?️ 2

cols = len(my_arr[0])
print(cols)  # ?️ 3

total_elements = rows * cols
print(total_elements)  # ?️ 6

在 Python 中获取二维数组的长度

如果我们需要获取 numpy 数组的长度，请向下滚动到下一部分。

len() 函数返回对象的长度（项目数）。

函数采用的参数可以是序列（字符串、元组、列表、范围或字节）或集合（字典、集合或冻结集合）。

我们首先将整个列表传递给 len() 函数以获取行数（嵌套列表的数量）。

假设每个嵌套列表具有相同数量的元素，我们可以通过访问嵌套列表并将结果传递给 len() 函数来获取列的长度。

我们可以通过将行数乘以列数来获得 2D 列表中的项目总数。

如果嵌套列表的大小可能不同，请使用 sum() 函数。

my_arr = [[1, 2, 3], [4, 5, 6, 7, 8]]

result = sum(len(l) for l in my_arr)

print(result)  # ?️ 8

sum 函数接受一个可迭代对象，从左到右对其项求和并返回总数。

我们将每个嵌套列表传递给 len 函数并计算总和。

如果您需要获取 2D numpy 数组的长度，请使用 shape 属性。

import numpy as np

my_2d_arr = np.array([[1, 2, 3], [4, 5, 6]])

print(my_2d_arr.shape)  # ?️ (2, 3)
print(my_2d_arr.size)  # ?️ 6


rows, cols = my_2d_arr.shape

result = rows * cols
print(result)  # ?️ 6