如何基于用户输入创建float类型的numpy矩阵?
转换为float类型numpy矩阵的实现方法
现有代码生成的int类型嵌套列表,可以通过以下两种方式实现转换:
- 方案一:直接对现有列表做类型转换
不需要修改之前的输入逻辑,在生成matrix列表后调用np.array()时指定dtype参数为浮点类型即可:
如果你需要单精度浮点可以把# 接你原有代码的最后部分 float_matrix = np.array(matrix, dtype=float) # 可自行验证结果 print(float_matrix) print("矩阵数据类型:", float_matrix.dtype)dtype=float替换为dtype=np.float32,默认的float对应双精度float64。 - 方案二:输入阶段直接处理为浮点型
可以修改输入时的类型转换逻辑,从转int改为转float,省去后续额外指定类型的步骤:import numpy as np m = int(input('Number of lines m: ')) matrix = [] for i in range(m): # 将map的第一个参数从int改为float,输入值直接转为浮点型存入列表 row = list(map(float, input(f'Line {i} ').split())) matrix.append(row) float_matrix = np.array(matrix) print(float_matrix)
内容的提问来源于stack exchange,提问作者Patrícia
相关产品推荐
相关产品推荐

