如何用更少量的代码编写这些函数？

Posted 2023-02-22

【中文标题】如何用更少量的代码编写这些函数？

【英文标题】：How to write these functions in smaller amount of code?

【发布时间】：2022-01-21 16:27:33

【问题描述】：

我有一长串汽车开发商和相关数据需要导入 Python，并且必须创建一个名为“制造商”的新列，对应于每个制造商。我怎样才能更简洁地做到这一点

import zipfile

zf = zipfile.ZipFile(r"C:\Users\aksel\Downloads\Cars.zip") 

zf.namelist() 
 
audi = pd.read_csv(zf.open("Audi.csv"))
bmw = pd.read_csv(zf.open("BMW.csv"))
ford = pd.read_csv(zf.open("Ford.csv"))

audi["manufacturer"] = "audi"
bmw["manufacturer"] = "bmw"
ford["manufacturer"] = "ford"


cars = pd.concat([audi, bmw, ford, hyundai, mercedes, skoda, toyota, vw])

【参考方案1】：

不要使用单独的变量，使用列表和循环。

cars_list = []
for filename in zf.namelist():
    car = pd.read_csv(zf.open(filename))
    car["manufacturer"] = filename.replace(".csv", "").lower()
    cars_list.append(car)

cars = pd.concat(cars_list)