Python二维数组提取指定行数据
要提取二维数组中的指定几行数据,可以使用切片 (slice) 的方式来实现。以下是一个示例代码:
def extract_rows(arr, start, end):
return arr[start:end+1]
# 示例二维数组
arr = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12]]
# 提取第2行到第3行的数据
start_row = 1
end_row = 2
extracted_rows = extract_rows(arr, start_row, end_row)
# 输出提取的结果
for row in extracted_rows:
print(row)
输出结果为:
[4, 5, 6]
[7, 8, 9]
在这个示例中,我们定义了一个 extract_rows 函数,它接受一个二维数组 arr,以及要提取的起始行 start 和终止行 end 作为参数。函数内部使用切片 arr[start:end+1] 来提取指定的行数据,并将结果返回。
在主程序中,我们使用示例二维数组 arr,并调用 extract_rows 函数提取第2行到第3行的数据。然后,我们使用一个循环遍历提取的结果,并打印每一行的数据。
原文地址: https://www.cveoy.top/t/topic/p6TQ 著作权归作者所有。请勿转载和采集!