📜  在 python 中查找二维列表形状的代码(1)

📅  最后修改于: 2023-12-03 15:23:16.468000             🧑  作者: Mango

在 Python 中查找二维列表形状的代码

在 Python 中,我们可以通过一些内置函数来查找二维列表的形状,即行数和列数。下面介绍两种常见的方式。

使用NumPy库

NumPy是Python中用于数值计算的核心库之一,可以处理高维数组。我们可以使用NumPy的shape属性来获取二维列表的形状。

import numpy as np

# 定义一个二维列表
arr = [[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]]

# 将二维列表转化为NumPy数组
np_arr = np.array(arr)

# 利用shape方法获取形状
shape = np_arr.shape

# 打印结果
print("该二维列表形状为:{}行{}列".format(shape[0], shape[1]))

代码解释:

  1. 首先,我们导入NumPy库。
  2. 然后,定义一个二维列表arr。
  3. 利用NumPy的array方法将其转化为NumPy数组。
  4. 利用NumPy数组的shape属性获取形状。
  5. 最后,打印结果。
使用Python内置函数

我们也可以使用Python的内置函数len()来获取行数和列数。

# 定义一个二维列表
arr = [[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]]

# 获取行数和列数
rows = len(arr)
cols = len(arr[0]) if rows > 0 else 0

# 打印结果
print("该二维列表形状为:{}行{}列".format(rows, cols))

代码解释:

  1. 首先,我们定义一个二维列表arr。
  2. 利用len()函数获取行数。
  3. 判断二维列表是否为空,如果不为空,利用len()函数获取第一行的元素个数,即列数;否则列数为0。
  4. 最后,打印结果。

以上两种方式都可以有效地获取二维列表的形状。如果你更喜欢NumPy库,就可以选择第一种方式;如果你更喜欢Python内置函数,就可以选择第二种方式。