python工作列表的方式

只是想看看如何分割字符串对象 data 在新行字符“\n”上,将结果存储在名为rows的变量中。-然后,使用print()函数显示前五个

f = open('crime_rates.csv', 'r')
data = f.read()

Print(data)  # few lines of data object

Albuquerque,749
Anaheim,371
Anchorage,828
Arlington,503
Atlanta,1379
Aurora,425
Austin,408
Bakersfield,542
Baltimore,1405
Boston,835

第一种方式:

rows = data.split('\n')
print(rows[0:5])

结果1:

['Albuquerque,749', 'Anaheim,371', 'Anchorage,828', 'Arlington,503', 'Atlanta,1379']

第二种方式,这显然不好。

f = open('crime_rates.csv', 'r')
data = f.read()
for rows in data:
    rows = data.split("\n")
    print(rows[0:5])
    break

结果2:

[“阿尔伯克基,749”,“阿纳海姆,371”,“安克雷奇,828”,“阿灵顿,503”,“亚特兰大,1379']

如何使用列表理解或lambda函数执行此操作:

print([[row[i] for row in data.split("\n")] for i in row[0:5]])