python - 在 python 中创建文件，指示整数是否在另一个列表中

Question

我有一个整数列表，比如说 [2, 4, 9]，然后我需要创建一个这样的 csv 文件:

1,0
2,1
3,0
4,1
5,0
6,0
7,0
8,0
9,1

基本上，对于每个整数，我需要检查整数是否在列表中，如果是我写1，否则是0。

我可以创建一个双列表，做一个 for 循环，检查整数是否在列表中，然后附加 (i,0) 或 (i,1) 取决于 i 是否在列表中，但我相信有一种更有效的方法可以做到这一点。这些列表很大，大约有十亿个整数。我查看了 numpy 中是否有可以帮助我的东西，但找不到任何东西。

Answer 1

使用list comprehensions创建数据并使用模块 csv写入 csv 文件。

s = set([2, 4, 9])
lists = [(i, 1) if i in s else (i, 0) for i in range(1, 10)]

# write to a csv file
import csv
with open('test.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerows(lists)

---

由于你的数据比较大，最好边创建边写入文件，如下图。

import csv

s = set([2, 4, 9])
with open('test.csv', 'w') as f:
    writer = csv.writer(f)

    for i in range(1, 10000000):
        t = (i, 1) if i in s else (i, 0)
        writer.writerow(t)