Dataframe获取指定列并将其单独保存

甜度超标°

已于 2022-08-23 19:08:20 修改

阅读量2.2k

点赞数

分类专栏：学习笔记文章标签： python pandas 机器学习 dataframe

于 2022-08-23 19:05:41 首次发布

本文链接：https://blog.csdn.net/qq_38703529/article/details/126491023

版权

学习笔记专栏收录该内容

83 篇文章 6 订阅

订阅专栏

Dataframe获取指定列并将其单独保存

问题描述
解决办法

问题描述

根据项目需要，要把表格拆分成两部分，一部分做训练一部分做测试。

解决办法

在jupyter notebook实现。

import pandas as pd
import os
from sklearn import datasets
from pandas import DataFrame
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

data = pd.read_csv('./6.csv', sep=';',index_col='datetime', parse_dates=True).drop(['changepoint'], axis=1)
data

在这里插入图片描述

df_train = data.iloc[:550].drop(['anomaly'], axis=1)
df_test = data.iloc[550:]
df_train.to_csv('./generated_' + '/train.csv')
df_train

在这里插入图片描述

df_test.to_csv('./generated_' + '/test.csv')
df_test

在这里插入图片描述
拆分保存的表格如下。

在这里插入图片描述
完整代码:

import pandas as pd
import os
from sklearn import datasets
from pandas import DataFrame
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

data = pd.read_csv('./6.csv', sep=';',index_col='datetime', parse_dates=True).drop(['changepoint'], axis=1)

df_train = data.iloc[:550].drop(['anomaly'], axis=1)
df_test = data.iloc[550:]

df_train.to_csv('./generated_' + '/train.csv')
df_test.to_csv('./generated_' + '/test.csv')