for i in list_1:
if i == item_1:
count_num += 1
return count_num
movieDataFile = ‘dataFiles/IMDB-Movie-Data.csv’
mov_df = pd.read_csv(movieDataFile)
print(mov_df.head(1))
print(mov_df[‘Genre’])
获得电影分类列表
temp_type_list = mov_df[‘Genre’].str.split(‘,’).tolist()
print(temp_type_list)
所有分类的列表(重复)
type_list = [i for j in temp_type_list for i in j]
print(type_list)
所有分类的列表(不重复)
genre_list = list(set(type_list))
print(genre_list)
分类对应的数量
genre_count_list = []
for item in genre_list:
item为分类名,由于一部电影不会有两个一样的分类名,直接计数即可
genre_count_list.append(CountItemNum(item, type_list))
print(genre_count_list)