仅从 numpy 数组中获取整数

Question

我正在处理数据，我有 phi、lambda 和 Z，看起来像这样：

18.49528 88.29922 2.955
19.24444 103.70188 528.784
24.79509 107.41260 36.138
24.80513 107.40091 33.336

我读取文件，然后使用名称提取数据

import numpy as np
file=np.loadtxt('data.txt')
phi=file[:,0]
lambda=file[:,1]
Z=file[:,2]

但是我必须解一些方程，并且在一个因素中，我应该只有 lambda 的部分整数，所以我这样做了

for i in lambda:
    print(int(i))
    A=(int(i))

当我打印 int(i) 时，我只得到部分整数，但是当我想将其保存在变量中时，这只保存一个值，而我想要 lambda 的所有值，但只得到部分整数，我该如何修复是吗？

Answer 1

我相信您正在尝试将

lambda

列表中的所有整数部分存储到一个单独的变量中。你做错的是你只是将

int(i)

值存储到

A

并在每次迭代中修改它，而不是将其存储到像列表或元组这样的集合中。

解决方法如下：

lambda_integers = []
for i in lambda_vals:
    print(int(i))
    lambda_integers.append(int(i))

另外，请不要使用

lambda

作为变量名。它可能与 Python 关键字

lambda

冲突，作为一个好的做法，您应该避免给出这样的变量名称。我已将

lambda

替换为

lambda_vals

。

您还可以使用列表理解来实现您的目的：

lambda_integers = [int(i) for i in lambda_vals]

希望有帮助。