csv中的字符串替换

Question

下面，我正在尝试替换 csv 中的数据。该代码有效，但它替换了文件中与 stocklevel 匹配的任何内容。

def updatestocklevel(quantity, stocklevel, code):
    newlevel = stocklevel - quantity
    stocklevel = str(stocklevel)
    newlevel = str(newlevel)
    s = open("stockcontrol.csv").read()
    s = s.replace (stocklevel ,newlevel) #be careful - will currently replace any number in the file matching stock level!
    f = open("stockcontrol.csv", 'w')
    f.write(s)
    f.close()

我的 csv 看起来像这样；

34512340,1
12395675,2
56756777,1
90673412,2
12568673,3
22593672,5
65593691,4
98593217,2
98693214,2
98693399,5
11813651,85
98456390,8
98555567,3
98555550,45
98553655,2
96553657,1
91823656,2
99823658,2

在我的程序的其他地方，我有一个搜索代码（8 位数字）的函数 是否可以说，如果code 在csv 的行中，则替换第二列中的数据？ (数据[2])

Answer 1

当您调用s.replace (stocklevel ,newlevel) 时，所有出现的stocklevel 都将替换为newlevel 的值。

string.replace(s, old, new[, maxreplace]): 返回字符串 s 的副本 将所有出现的子字符串 old 替换为 new。如果可选 参数 maxreplace 给出，第一个 maxreplace 出现是 替换了。

source

按照您的建议，您需要获取 code 并使用它来替换库存水平。

这是一个示例脚本，它将 8 位代码和新的库存水平作为命令行参数并替换它：

import sys
import re
code = sys.argv[1]
newval= int(sys.argv[2]) 
f=open("stockcontrol.csv")
data=f.readlines()
print data
for i,line in enumerate(data):
  if re.search('%s,\d+'%code,line): # search for the line with 8-digit code
    data[i]  = '%s,%d\n'%(code,newval) # replace the stock value with new value in the same line
f.close()

f=open("in.csv","w")
f.write("".join(data))
print data
f.close()

使用 Python 的csv 模块的另一种解决方案：

import sys
import csv

data=[]
code = sys.argv[1]
newval= int(sys.argv[2])
f=open("stockcontrol.csv")
reader=csv.DictReader(f,fieldnames=['code','level'])
for line in reader:
  if line['code'] == code:
    line['level']= newval
  data.append('%s,%s'%(line['code'],line['level']))
f.close()

f=open("stockcontrol.csv","w")
f.write("\n".join(data))
f.close()

警告：在尝试这些脚本时备份输入文件，因为它们会覆盖输入文件。

如果您将脚本保存在名为 test.py 的文件中，则调用它：

python test.py 34512340 10.

这应该将代码 34512340 的 stockvalue 替换为 10。

Answer 2

为什么不使用好的旧正则表达式？

import re

code, new_value = '11813651', '885' # e.g, change 85 to 885 for code 11813651
print (re.sub('(^%s,).*'%code,'\g<1>'+new_value,open('stockcontrol.csv').read()))

Answer 3

由于它是一个 csv 文件，我建议使用 Python 的 csv 模块。您将需要写入一个新文件，因为读取和写入同一个文件会变得很糟糕。您可以随时重命名它。

此示例使用StringIO (Python 2) 将您的 csv 数据嵌入代码中并将其视为文件。通常你会打开一个文件来获取输入。

更新

import csv
# Python 2 and 3
try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO


CSV = """\
34512340,1
12395675,2
56756777,1
90673412,2
12568673,3
22593672,5
65593691,4
98593217,2
98693214,2
98693399,5
11813651,85
98456390,8
98555567,3
98555550,45
98553655,2
96553657,1
91823656,2
99823658,2
"""


def replace(key, value):
    fr = StringIO(CSV)
    with open('out.csv', 'w') as fw:
        r = csv.reader(fr)
        w = csv.writer(fw)

        for row in r:
            if row[0] == key:
                row[1] = value
            w.writerow(row)

replace('99823658', 42)