Python获取二维数组行列数的2种方法

Python获取二维数组行列数的2种方法

在Python中处理二维数组时，获取其行数和列数是常见的需求。本文将介绍两种常用的方法，并通过示例代码详细说明。

方法一：使用NumPy库的shape属性

NumPy是Python中用于科学计算的核心库，提供了强大的数组操作功能。通过NumPy的shape属性，可以方便地获取二维数组的行数和列数。

import numpy as np

x = np.array([[1,2,5],[2,3,5],[3,4,5],[2,3,6]])
# 输出数组的行和列数
print(x.shape) # (4, 3)
# 只输出行数
print(x.shape[0]) # 4
# 只输出列数
print(x.shape[1]) # 3

在上述代码中，我们首先导入了NumPy库，然后创建了一个二维数组x。通过x.shape属性，我们可以直接获取数组的形状，即行数和列数。shape属性返回一个元组，其中第一个元素表示行数，第二个元素表示列数。

方法二：使用内置len()函数

如果不使用NumPy库，我们可以利用Python内置的len()函数来获取二维数组的行数和列数。这种方法适用于普通的列表嵌套结构。

arr = [[1,4,7,10,15], [2,5,8,12,19], [3,6,9,16,22], [10,13,14,17,24]]

# 获取行数
rows = len(arr)
print(rows) # 4

# 获取列数
cols = len(arr[0])
print(cols) # 5

在上述代码中，我们定义了一个二维列表arr。通过len(arr)可以获取数组的行数，而len(arr[0])则可以获取第一行的元素个数，即列数。这种方法简单直接，适用于不需要复杂数组操作的场景。

总结

通过上述两种方法，我们可以灵活地获取Python中二维数组的行数和列数。NumPy的shape属性适用于科学计算和数据处理，而内置len()函数则适用于简单的列表操作。根据实际需求选择合适的方法，可以提高代码的效率和可读性。

以上就是Python获取二维数组行列数的2种方法的详细内容，更多关于Python二维数组的资料请关注本站其它相关文章！