使用 itertools.product 写入 csv 文件

Question

我已经创建了一个具有独特组合的脚本，但我仍然不知道如何将它写入 csv 文件。现在它只是在我的命令行中打印出来，很难掌握数据。

我的代码：

from itertools import combinations, product

crust = ['Thin Crust', 'Hand Tossed']
topping = ['Bacon', 'Pepperoni', 'Steak']
sauce = ['Tomato', 'BBQ', 'Ranch']

topping_combs = combinations(topping, r=2)
sauce_combs = combinations(sauce, r=2)

topping_combs = list(topping_combs)
sauce_combs = list(sauce_combs)

prod = product(crust, topping_combs, sauce_combs)
for c, t, s in prod:
    print(c, *t, *s, sep=', ')

Answer 1

这是代码。它将打开 csv 文件并在 for 循环的每次迭代中写入它。您将在每个单元格中看到一个项目，因为它是在终端中给出的。

from itertools import combinations, product
import csv
crust = ['Thin Crust', 'Hand Tossed']
topping = ['Bacon', 'Pepperoni', 'Steak']
sauce = ['Tomato', 'BBQ', 'Ranch']

topping_combs = combinations(topping, r=2)
sauce_combs = combinations(sauce, r=2)

topping_combs = list(topping_combs)
sauce_combs = list(sauce_combs)

prod = product(crust, topping_combs, sauce_combs)
combine_list=[]
with open('Book1.csv', 'w', newline='') as file:
    for c, t, s in prod:
        print(c, *t, *s, sep=', ')

        writer = csv.writer(file)
        writer.writerow([c,*t,*s])

使用 itertools.product 写入 csv 文件

Writing to csv file with itertools.product

csv

itertools

python-3.x

export-to-csv