给定一个csv数据集,其中某个属性为电影类型,比如某条数据的电影类型为“剧情/爱情”,利用python,将所有包含“剧情”的数据中,剧情修改为动作,其他内容不变
时间: 2023-06-25 16:05:18 浏览: 94
基于python实现影视作品数据分析功能实现源码
5星 · 资源好评率100%
可以使用pandas库读取csv数据,然后使用字符串的contains()方法判断某个属性中是否包含某个字符串,最后使用replace()方法替换该属性中的内容。
以下是示例代码:
```python
import pandas as pd
# 读取csv数据
df = pd.read_csv('movie_data.csv')
# 判断电影类型是否包含“剧情”,并将该属性中的“剧情”替换为“动作”
df.loc[df['电影类型'].str.contains('剧情'), '电影类型'] = df['电影类型'].str.replace('剧情', '动作')
# 保存修改后的数据
df.to_csv('modified_movie_data.csv', index=False)
```
在上述代码中,我们首先使用pandas库的read_csv()方法读取电影数据集,然后使用str.contains()方法判断电影类型是否包含“剧情”,并使用str.replace()方法将该属性中的“剧情”替换为“动作”。最后,使用to_csv()方法将修改后的数据保存到新的csv文件中。
阅读全文