<p>免责声明:我是Python新手,所以这可能在很多方面都是错误的</p>
<p>问题:我希望将csv上载到python,将数据拆分为测试和训练集,然后将测试和训练集导出到两个单独的csv文件中。我有以下代码:</p>
<pre><code>import pandas as pd
dataset = pd.read_csv('data.csv')
# Dividing dataset into label and feature sets
X = dataset.drop('Bankrupt?', axis = 1) # Features
Y = dataset['Bankrupt?'] # Labels
# Normalizing numerical features so that each feature has mean 0 and variance 1
feature_scaler = StandardScaler()
X_scaled = feature_scaler.fit_transform(X)
# Dividing dataset into training and test sets
X_train, X_test, Y_train, Y_test = train_test_split( X_scaled, Y, test_size = 0.3, random_state = 100)
#Export to csv
X_train = pd.DataFrame.to_csv(r'file path where csv is to be stored\training.csv', index = False)
X_test = pd.DataFrame.to_csv(r'file path where csv is to be stored\test.csv', index = False)
</code></pre>
<p>但我一直得到以下错误:“str”对象没有“to_frame”属性</p>
<p>我做错了什么,或者有没有更好的方法来达到同样的效果</p>