条件列表 Python

Question

我正在打开 zip 中存在的所有文件，然后将字符串转换为列表，然后将列表分成每行 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 = "文件 1 + 文件 2"

文件 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

代码用途：

我正在添加将此列表转换为字典，因此将键添加到每个条目为 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'}]

条件列表 Python

Conditional list Python

python

arrays

formatting

list

python-3.x