Python CSV对字段中的数据求平方根

Question

我有一个 CSV 文件，其中字段中包含数字，

我编写了一个脚本，尝试计算此 CSV 文件中每个字段中数字的平方根，

import sys, os

import csv, math
import tempfile
import shutil

element = []
filename = 'test.csv'
with open(filename, 'rb') as f, tempfile.NamedTemporaryFile(mode='wb', delete=False) as g:
    writer = csv.writer(g, delimiter = ',')
    for row in csv.reader(f):
        element = element in row
        row = math.sqrt(element)
        writer.writerow([row])

shutil.move(g.name, filename)

但是输出不是我想要的，

我应该编辑什么脚本？

Answer 1

您正在计算 False 的平方根，这与 0 的平方根相同

>>> element = element in row
False
>>> row = math.sqrt(element)
0

修改如下:

filename = 'test.csv'
with open(filename, 'rb') as f, tempfile.NamedTemporaryFile(mode='wb', delete=False) as g:
    writer = csv.writer(g, delimiter = ',')
    for row in csv.reader(f):
        row = [math.sqrt(float(num)) for num in row]
        writer.writerow(row)