将zip与两个列表一起使用

Question

我试图遍历两个列表，一个包含76个文件，另一个包含76个变量以读取文件。 我认为我可以使用zip（list1，list2）遍历它们，但是由于某些原因，它无法正常工作。 这是我在做什么：

list1=['file1', 'file2', 'file3']
list2=['v1','v2','v3']
for i,j in zip(list1,list2):
    j=pyfits.getdata(i)       #just trying to read a FITS file

当我用print做同样的事情时，它工作得很好：

list1=['file1', 'file2', 'file3']
list2=['v1','v2','v3']
for i,j in zip(list1,list2):
    print i, j

有任何想法吗？

Answer 1

j=pyfits.getdata(i)

该行不将数据存储到名称存储在j的变量。 而是将数据存储到变量j 。 您需要的是列出数据列表：

data = [pyfits.getdata(filename) for filename in list1]

这将创建一个列表，其中data[0]是来自第一个文件的data[1] ， data[1]是来自第二个文件的数据，依此类推。

Answer 2

我假设您正在尝试将list2的字符串分配给pyfits.getdata()返回的数组。 但是， list2包含字符串，因此无法以这种方式将数组分配给字符串。 有一些骇人听闻的方法可以做到这一点，但这在Python中并不是一个好习惯。 我建议使用字典，其中list2中的字符串可以是字典中的键。

这是一个例子：

data = {}
list1=['file1', 'file2', 'file3']
list2=['v1','v2','v3']
for i,j in zip(list1,list2):
    data[j] = pyfits.getdata(i)

然后，您可以使用data[v1] ， data[v2]等访问数据。

Answer 3

您可以使用map功能：

list2 = map(pyfits.getdata,list1)
zip(list1,list2) #not needed here

然后创建漂亮的字典

di = dict(zip(list1,list2))

di结构类似于{filename1:data1,filename2,data2}