条件列表 Python

Question

我打开 zip 中存在的所有文件，然后将字符串转换为 List，然后将列表从每行分成 5 个元素。

我的代码：

with zipfile.ZipFile(dummy.zip, 'r') as mz:
    for f in mz.namelist():
        data = mz.read(f).decode('utf-8').replace("\n", '').replace("\r", ',').lstrip().rstrip().split(",") #Removing all spaces and \r from text.
        allist = [data[i:i + 5] for i in range(len(data))[::5]] #breaking list into 5 elements each

如果所有值都存在，我能够完美地做到这一点，但是如果特定行上缺少任何元素，那么它会从第二行中获取值，但我想在缺少元素的位置添加NA 。

文件数据： Dummy.zip = "File1 + File2"

文件1：

Yuli, yu@test.com, 0001, 8902
Su, su@test.com, 0002, 8903, Manager

文件：2

Zaon, zn@test.com, 100, 9087, Analyst
June, ju@test.com, 278, 6078

代码的目的：

我正在添加将此列表转换为 dict，因此将键添加到每个条目，如name, email, ext, Empid, Level

Answer 1

你可以这样做：

texts = [b"Yuli, yu@test.com, 0001, 8902\r\nSu, su@test.com, 0002, 8903, Manager",
         b"Zaon, zn@test.com, 100, 9087, Analyst\r\nJune, ju@test.com, 278, 6078"]

result =[]
keys = "name,email,ext,Empid,Level".split(",")
# you do: 
# with zipfile.ZipFile(dummy.zip, 'r') as mz: 
#     for f in mz.namelist(): 
#         t = mz.read(f) here instead
# instead
for t in texts:
    for line in t.decode("utf-8").split("\n"):
        d = {k:"NA" for k in keys} # init all keys with NA
        # update with actual values
        d.update(zip(keys,(l.strip() for l in line.strip().split(","))))
        # add to result list
        result.append(d) 
print(result)

输出：

[{'name': 'Yuli', 'email': 'yu@test.com', 'ext': '0001', 'Empid': '8902', 'Level': 'NA'}, 
 {'name': 'Su', 'email': 'su@test.com', 'ext': '0002', 'Empid': '8903', 'Level': 'Manager'},
 {'name': 'Zaon', 'email': 'zn@test.com', 'ext': '100', 'Empid': '9087', 'Level': 'Analyst'}, 
 {'name': 'June', 'email': 'ju@test.com', 'ext': '278', 'Empid': '6078', 'Level': 'NA'}]