为什么在循环中无法修改列表内numpy数组的数据类型?
失效原因
np.astype()方法本身不会修改原数组,只会返回一个完成类型转换的全新数组- 循环中获取的
y是列表元素引用的临时副本,直接对y重新赋值只会修改临时变量的指向,不会改动arrays列表中存储的原有引用,因此原列表里的复数数组没有被替换,看不到转换效果。
可行修改方案
你可以通过索引直接修改原列表的元素,替换为转换后的新数组:
import numpy as np a=np.array([2+3j,3+4j]) b=np.array([1+2j,4+3j]) arrays=[a,b] for i in range(len(arrays)): arrays[i] = arrays[i].astype('float64') print(arrays)
也可以用更简洁的列表推导式直接生成转换后的新列表:
arrays = [arr.astype('float64') for arr in arrays]
补充说明:numpy将复数数组转换为float64类型时,默认会直接丢弃虚部保留实部,符合你的需求,不需要额外加处理逻辑。
内容的提问来源于stack exchange,提问作者user16424912
相关产品推荐
相关产品推荐

