python：创建校验位函数

Question

我正在尝试创建校验位并将它们附加到原始 UPC 之后。这是sample data

因为有前导0，所以我必须先将数据读取为字符串：
import pandas as pd upc = pd.read_csv("/Users/lee/Desktop/upc.csv", dtype = str)

以下是校验位算法的示例：
如果 upc 为 003459409000
步骤（1）0 + 3*0 + 3 + 3*4 + 5 + 3*9 + 4 + 3*0 + 9 + 3*0 + 0 + 3*0 = 60
步骤（2）60 mod 10 = 0
步骤（3）check digit = 0 (if it's not 0, then check digit = 10 - number in step 2)

根据算法，代码如下：

def add_check_digit(upc_str):  
upc_str = str(upc_str)
if len(upc_str) != 12: 
raise Exception("Invalid length")

odd_sum = 0
even_sum = 0 
for i, char in enumerate(upc_str): 
j = i+1 
if j % 2 == 0: 
even_sum += int(char) 
else:m 
odd_sum += int(char) 
total_sum = (even_sum * 3) + odd_sum 
mod = total_sum % 10 
check_digit = 10 - mod 
if check_digit == 10: 
check_digit = 0 
return upc_str + str(check_digit) 

如果我运行此代码，它会提供正确的校验位并将此结果附加到原始 UPC 的末尾。对于上面的示例，如果我输入：

add_check_digit('003459409000')

输出给出 13 位 UPC 0034594090000。

现在我的问题是：

1. 此功能仅适用于单个 upc，即我必须复制/粘贴每个单个 upc 并获取校验位。如何创建适用于数据框中的 UPS 列表的函数？每个结果应返回一个 13 位 UPC，并在原始 UPC 之后附加校验位。

2. UPC 被读取为字符串。如何将该功能应用于 UPC？我想我应该以某种方式将字符串转换为数字。我真的是 python 新手。

3. 获得新的 UPC 后，如何将结果保存到 csv 文件中？

非常感谢您的帮助。

Answer 1

为我设置了数据，因为我没有 CSV 文件，以下步骤与您的相同

df = pd.read_csv("/Users/lee/Desktop/upc.csv", dtype = str)

数据设置

import pandas as pd
df=pd.DataFrame({"upc_in_file":['003459409000','003459409001','003459409002']})

def add_check_digit(upc_str):  
    upc_str = str(upc_str)
    if len(upc_str) != 12: 
        raise Exception("Invalid length")

    odd_sum = 0
    even_sum = 0 
    for i, char in enumerate(upc_str): 
        j = i+1 
        if j % 2 == 0: 
            even_sum += int(char) 
        else:
            odd_sum += int(char) 
            total_sum = (even_sum * 3) + odd_sum 
            mod = total_sum % 10 
            check_digit = 10 - mod 
        if check_digit == 10: 
            check_digit = 0 
    return upc_str + str(check_digit) 

将上述函数应用于 upc 列（从文件中读取的列）

df['new_upc']=df['upc_in_file'].apply(add_check_digit)

现在保存文件！

df.to_csv("my_updated_upc.csv")

这看起来像