如何通过从另一个CSV文件查找来替换CSV列中的值？

jpfvwuh4 于 2023-01-15 发布在其他

关注(0)|答案(1)|浏览(141)

我有两个单独的CSV文件。我想扫描这两个CSV文件并查找匹配的SKU编号。如果它们确实匹配，则根据表1中与SKU编号相关的编号更新表2中的编号。

import csv
  
  with open("C:\\Users\\Owner\\OneDrive\\Desktop\\Test\\read.csv", 'r') as file:
  # Create a CSV reader
  reader = csv.reader(file)
  rows1 = [row for row in reader]

  with  open("C:\\Users\\Owner\\OneDrive\\Desktop\\Test\\import.csv", 'r') as file:
  reader = csv.reader(file)
  rows2 = [row for row in reader]

  column1_values = [row[7] for row in rows1]
  column2_values = [row[8] for row in rows2]
  column3_values = [row[8] for row in rows1]

  for row in rows1:
      value2 = row[7]

  for row in rows2:
      value1 = row[8]

  if value1 in column1_values and value2 in column2_values:
    value3 = column3_values
    print(value3)

到目前为止，我有python代码能够读取这两个文件。但是我在匹配sku编号，然后在之后更新库存编号方面遇到了麻烦。
我要从中提取数据的第一个文件：

,,kh10,,,,0
,,kh12,,,,21
,,kh13,,,,1
,,kh11,,,,45
,,kh20,,,,26

要更新的第二个文件：

,kh20,,,0
,kh16,,,47
,kh12,,,31
,kh10,,,2
,kh13,,,0

我想匹配那些KH数字，然后从第一个文件中提取第一个文件中最后一列的数字，并用这些数字替换第二个文件中的最后一列。

csv

来源：https://stackoverflow.com/questions/75072579/how-to-replace-values-in-a-csv-column-by-looking-up-from-another-csv-file

1条答案

按热度按时间

t2a7ltrp1#

您希望创建一个 dictionary，其中键是SKU编号，值是您希望包含在结果中的数量。

with open("C:\\Users\\Owner\\OneDrive\\Desktop\\Test\\read.csv", 'r') as file:
    # Create a CSV reader
    reader = csv.reader(file)
    inventory = {row[2].strip(): int(row[7]) for row in reader}

这将创建一个inventory字典，如下所示：

{'kh10': 0, 'kh12': 21, 'kh13': 1, 'kh11': 45, 'kh20': 26}

现在，当您读取另一个文件时，将其E列替换为以下dict中的相应值：

with open("C:\\Users\\Owner\\OneDrive\\Desktop\\Test\\import.csv", 'r') as file:
    reader = csv.reader(file)
    rows2 = [] # Create an empty list to hold all rows
    for row in reader:
        sku = row[1]
        current_inventory = int(row[4]) # Current value

        # Get updated value, keep current value if it doesn't exist in lookup dict
        updated_inventory = inventory.get(sku, current_inventory)

        # Store updated value
        row[4] = updated_inventory

        # Append row to master list
        rows2.append(row)

现在，您有了一个包含第二个CSV文件中的数据的列表，但是更新了库存编号。

[['', 'kh20', '', '', 26],
 ['', 'kh16', '', '', 47],
 ['', 'kh12', '', '', 21],
 ['', 'kh10', '', '', 0],
 ['', 'kh13', '', '', 1]]

您可能已经知道如何将其写入CSV文件。

赞(0）回复(0）举报 2023-01-15

我来回答

如何通过从另一个CSV文件查找来替换CSV列中的值？

1条答案

相关问题

热门标签

最新问答