在Python中输入数组可以通过以下几种方法:
一维数组输入
1. 使用`input()`函数获取用户输入,然后使用`split()`函数将输入的字符串按空格或其他分隔符分割成列表。
2. 将分割后的列表中的每个元素转换为适当的数值类型(如`int`或`float`)。
示例代码:
获取用户输入
arr_input = input("请输入一维数组的元素,用空格隔开:")
分割输入的字符串
arr_list = arr_input.split()
将字符串列表转换为整数列表
arr = [int(n) for n in arr_list]
print(arr)
二维数组输入
1. 首先输入数组的行数和列数。
2. 然后逐行输入数组的元素,每行元素用空格隔开。
3. 将每行的字符串分割并转换为整数列表,最后将这些列表组合成二维数组。
示例代码:
获取行数和列数
rows = int(input("请输入二维数组的行数:"))
columns = int(input("请输入二维数组的列数:"))
初始化二维数组
array_2d = [*columns for _ in range(rows)]
逐行输入并填充二维数组
for i in range(rows):
array_2d[i] = list(map(int, input("请输入第{}行,用空格隔开的元素:".format(i+1)).split()))
print(array_2d)
使用`numpy`库
1. 导入`numpy`库。
2. 使用`numpy.array()`函数将列表转换为`numpy`数组。
示例代码:
import numpy as np
获取用户输入
arr_input = input("请输入一维数组的元素,用逗号分隔:")
分割输入的字符串
arr_list = arr_input.split(',')
将字符串列表转换为整数列表
arr = np.array(arr_list, dtype=int)
print(arr)
以上方法可以帮助你在Python中输入一维和二维数组。