Python CSV在字段中取数据的平方根

亨氏

我有一个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)

但是输出不是我想要的,

在此处输入图片说明

我应该怎样编辑脚本?

扬·维尔辛斯基

您正在计算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)

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

TOP 榜单

热门标签

归档