我被这部分代码困住了。我无法在 Python 中的 .csv 中拆分一行。我的代码是:
import csv
# Get movie titles
movies={}
with open(path+'/ratings.csv') as csvfile:
reader = csv.reader(csvfile, delimiter=',')
for row in reader:
(Id,title)=row.split(',')[0:1]
Run Code Online (Sandbox Code Playgroud)
所以我想获取电影的 ID,即第一个值,而它们的标题是第二个值。.csv 文件如下所示:
userId,movieId,rating,timestamp
1,31,2.5,1260759144
1,1029,3.0,1260759179
1,1061,3.0,1260759182
1,1129,2.0,1260759185
1,1172,4.0,1260759205
1,1263,2.0,1260759151
1,1287,2.0,1260759187
1,1293,2.0,1260759148
1,1339,3.5,1260759125
1,1343,2.0,1260759131
1,1371,2.5,1260759135
1,1405,1.0,1260759203
1,1953,4.0,1260759191
....
Run Code Online (Sandbox Code Playgroud)
但由于某种原因,它给了我这个错误:
AttributeError: 'list' 对象没有属性 'split'
当我搜索时,其他人也遇到了类似的问题,但主要是因为他们是 .split 文件不是行,但我正在拆分行...
row已经是一个列表,这就是csv模块的重点(因为您将逗号指定为分隔符,您可以省略它,因为它是默认值)
做就是了:
Id,title = row[0:1]
Run Code Online (Sandbox Code Playgroud)
顺便说一句,next(reader)在for循环之前跳过标题行。