Python CSV將單元從一個csv文件復制到另一個

Question

編輯：這已經解決。 我已將解決方案添加到本文的底部。

我有兩個文件，newStock.csv和sortedBlad.csv。

newStock.csv看起來像

SKU_IN,QTY_IN
3201MO-1,1
320205-1,4

sortedBlad.csv看起來像

eid,sku,asd,asdf,Asdg,QTY,sdfasf,asdf,sdfsf,dfsdf,sdfs,dddd,dssss,Name,dddddpo,odommm,mdmdjkon,kkkkknn,okokokok,okokokb,jj,mb,bgtio,

1.11E+11,3201MO-1,,,,5,55,,£20.90,08-Aug-14,,New,9,FORD FOCUS MK1 98 04  OIL SUMP PAN FOR ENGINE  1.8 16V 2.0 16V  & ST RS PETROL,Oil Sumps,174039,,3,27-Jan-15 11:44:30 GMT,,,1000,FALSE,

我目前擁有的代碼通過sortedBlad.csv並復制eid，sku和name列，並將它們插入另一個文件中。 它還會匹配newStock中的草圖並將它們附加到每行的末尾。 我面臨的問題是，我似乎無法從新庫存中獲取數量，也無法將其附加到每行的末尾。 當前代碼的輸出如下。

    1.11E+11,3201MO-1,5,FORD FOCUS MK1 98 04  OIL SUMP PAN FOR ENGINE  1.8 16V 2.0 16V  & ST RS PETROL,3201MO-1,,
    1.21E+11,320205-1,6,FORD FOCUS 2004  2008 FRONT BUMPER  RADIATOR GRILLE BONNET WITH CHROME TRIM,320205-1,,

您會看到該sku被追加到該行的末尾。 我還要在sku之后附加newStock數量。

我到目前為止的代碼如下。

import csv

test_file = "sortedBlad.csv"
test_file2 = "sortedNewStock.csv" 

test_file2_array = []

finalFile = "final.csv"

csv_f_array = []

csv_file = csv.DictReader(open(test_file, 'r'), delimiter=',', quotechar='"')
csv_file2 = csv.DictReader(open(test_file2, 'r'), delimiter=',', quotechar='"')


with open(test_file2, 'rt') as a:
        reader = csv.reader(a, delimiter=",")
        for row in reader:
            test_file2_array.append(row[0])

#print(test_file2_array)

file = open(finalFile, 'w')

for eachline in csv_file:
    for SKU_IN in test_file2_array:
        if SKU_IN == eachline['sku']:
            print(SKU_IN+'-'+test_file2_array[1])
            # why wont you write the qty?
         file.write(eachline['eid']+','+eachline['sku']+','+eachline['QTY']+','+eachline['Name']+','+SKU_IN+','+',\n')

file.close

謝謝你們的答復。 我設法通過在reddit中的learningpython irc的某人的幫助下解決了該問題。

解決方案是使用字典將SKU_IN作為鍵存儲，將QTY_IN作為值存儲。 然后在寫文件時調用它。

Answer 1

您是否考慮過使用與表格數據集本機兼容的程序包來支持所需的操作，例如pandas ？

import pandas as pd

f1=pd.read_csv("sortedBlad.csv")
f2=pd.read_csv("sortedNewStock.csv")

#Filter columns in f1
f1sub=f1[ ['eid', 'sku', 'name' ] ]

#Merge data
mergef = pd.merge(f1sub,f2,left_on='sku',right_on='SKU_IN')

合並功能將自動為您處理對齊。

Answer 2

您只將sortedNewStock.csv （ SKU_IN ）的第一列附加到test_file2_array ：

test_file2_array.append(row[0])

您還需要從該文件中獲取QTY_IN列。 我建議以下內容同時添加SKU_IN和QTY_IN ：

test_file2_array.append(row[0:2])

然后改變

for SKU_IN in test_file2_array:

至

for SKU_IN, QTY_IN in test_file2_array:

因此，您的代碼如下所示：

.
.
.
with open(test_file2, 'rt') as a:
    test_file2_array = [row[0:2] for row in csv.reader(a, delimiter=",")]

#print(test_file2_array)

file = open(finalFile, 'w')

for eachline in csv_file:
    for SKU_IN, QTY_IN in test_file2_array:
        if SKU_IN == eachline['sku']:
            print(SKU_IN + '-' + QTY_IN)
            file.write(','.join((eachline['eid'], eachline['sku'],
                                 eachline['QTY'], eachline['Name'],
                                 SKU_IN, QTY_IN, '\n')))

Answer 3

您不會在任何地方保留來自newStock.csv的QTY_IN信息。

import csv

blad_file = "blad.csv"
newStock_file = "stock.csv" 

newStock_file_array = []

finalFile = "final.csv"

csv_blad = csv.DictReader(open(blad_file, 'r'), delimiter=',', quotechar='"')
csv_newStock = csv.DictReader(open(newStock_file, 'r'), delimiter=',', quotechar='"')

for line in csv_newStock:
    newStock_file_array.append(line)

with open(finalFile, 'w') as file_out:
    for eachline in csv_blad:
        for newStock in newStock_file_array:
            if newStock['SKU_IN'] == eachline['sku']:
                # print(newStock['SKU_IN']+'-'+newStock['QTY_IN'])
                file_out.write(eachline['eid']+','+eachline['sku']+','+eachline['QTY']+','+eachline['Name']+','+newStock['QTY_IN']+','+'\n')