Save the Titanic dataframe as a pickle file in the same folder
时间: 2024-10-03 12:02:49 浏览: 33
knn-class-1-titanic.rar_The Class_kNN c code_titanic
如果你想把Titanic数据框保存为pickle文件,可以按照以下Python代码步骤操作:
假设你的DataFrame已经被命名为`titanic_df`,并且你想要保存在当前工作目录(默认为与脚本相同的文件夹):
```python
import pandas as pd
import pickle
# 加载Titanic数据(如果还没有加载)
# titanic_df = pd.read_csv('titanic_data.csv') # 假设数据是从CSV文件读取的
# 或者直接使用内置的dataframe,例如:
titanic_df = ... # 请提供实际的数据框内容
# 将数据帧保存为pickle文件
filename = 'titanic.pkl' # 指定文件名
with open(filename, 'wb') as handle:
pickle.dump(titanic_df, handle)
# 现在,pickle文件已经在同一文件夹中了
```
这段代码首先导入pandas用于处理数据,以及pickle库用于序列化和反序列化数据。然后,将`titanic_df`对象写入到名为'titanic.pkl'的pickle文件中,'wb'表示写二进制模式。
阅读全文