python - 使用 'for' 或 'if' 条件在 Python 中创建 CSV 文件

Question

我设计的代码如下:

import csv
import numpy as np
data = [['Diameter', 'color', 'no']]
with open('samp1.csv', 'w') as f:
    writer = csv.writer(f, delimiter=',')
    for row in data:
        writer.writerow(row)
    for i in np.arange(20,30,0.2):
        writer.writerow(i)

f.close()

我想在直径列中以0.2的增量保存20-30之间的数字，但它给出了错误，甚至不保存CSV文件。有人可以提出任何解决方案吗？即使其他列也有不同的范围，所以我需要相同的方法来处理该代码。

CSV 示例

diameter    color     number
20          2         3
20          2.5       3
20          3         3
20          3.5       3
20.2        2         3
20.2        2.5       3
20.2        3         3
20.2        3.5       3
.
.
.
.
22          2         4
22          2.5       4
22          3         4
22          3.5       4
22.2        2         4
22.2        2.5       4
22.2        3         4
22.2        3.5       4

Answer 1

我想这可以帮助你:

import csv

data = ['Diameter', 'color', 'no']
numb = [i for i in range(20,30,2)]
print(numb)
with open('samp1.csv', 'w') as f:
    writer = csv.writer(f, delimiter=',')
    writer.writerow(data)
    for i in numb:
            writer.writerow([i,i,i])

f.close()