python - Error =, str, el objeto no tiene atributo, to_frame,

Descargo de responsabilidad: soy nuevo en Python, por lo que probablemente esto esté mal en muchos sentidos...

Pregunta: Deseo cargar un csv en Python, dividir los datos en un conjunto de prueba y entrenamiento y luego exportar los conjuntos de prueba y entrenamiento en dos archivos csv separados. Tengo el siguiente código:

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)

Pero sigo recibiendo el siguiente error: el objeto 'str' no tiene el atributo 'to_frame'

¿Hay algo que esté haciendo mal o hay alguna forma mejor de lograr el mismo resultado?

2

Por favor, publique el seguimiento de la pila la próxima vez, indicando en qué líneaSe origina el error.

Tomás

27/03/2021 a las 14:39

------------------------------------

to_csv es un método en DataFrame, así que creo que te refieres a dataset.to_csv en lugar de pd.DataFrame.to_csv en las dos últimas líneas.

------------------------------------

pd.Dataframe es la clase de marco de datos, no una instancia. to_csv pretende ser un método de instancia, llamado desde una instancia. Si lo llama como un método de clase (como lo hizo), su primer argumento debe ser un marco de datos.e.

Cualquiera de estos debería funcionar:

dataset.to_csv(r'file path ...)   # This is the intended use
pd.DataFrame.to_csv(dataset, r'file path ...)   # Harder to read and more prone to error