怎么将数据集ObesityDataSet_raw_and_data_sinthetic.csv中Gender为Female和Male的数据分别提取出来
时间: 2023-11-12 10:07:41 浏览: 52
提取数据集ObesityDataSet_raw_and_data_sinthetic.csv中Gender为Female和Male的数据,可以使用pandas库中的DataFrame提取方法,具体步骤如下:
1. 首先,导入pandas库并读取数据集ObesityDataSet_raw_and_data_sinthetic.csv。
```python
import pandas as pd
df = pd.read_csv('ObesityDataSet_raw_and_data_sinthetic.csv')
```
2. 然后,使用DataFrame的loc()方法提取Gender列为Female或Male的数据,并将结果保存到新的DataFrame中。
```python
female_df = df.loc[df['Gender'] == 'Female']
male_df = df.loc[df['Gender'] == 'Male']
```
这样就可以将Gender为Female和Male的数据分别提取出来并保存到新的DataFrame中。如果需要将这些数据保存到文件中,可以使用to_csv()方法,如下所示:
```python
female_df.to_csv('female.csv', index=False)
male_df.to_csv('male.csv', index=False)
```
完整的代码如下:
```python
import pandas as pd
df = pd.read_csv('ObesityDataSet_raw_and_data_sinthetic.csv')
female_df = df.loc[df['Gender'] == 'Female']
male_df = df.loc[df['Gender'] == 'Male']
female_df.to_csv('female.csv', index=False)
male_df.to_csv('male.csv', index=False)
```
希望能对你有所帮助!