diff --git a/spec/API_specification/dataframe_api/dataframe_object.py b/spec/API_specification/dataframe_api/dataframe_object.py index 9a727cf3..6460eb07 100644 --- a/spec/API_specification/dataframe_api/dataframe_object.py +++ b/spec/API_specification/dataframe_api/dataframe_object.py @@ -199,7 +199,7 @@ def insert_column(self, column: Column[Any]) -> DataFrame: .. code-block:: python new_column = df.get_column_by_name('a') + 1 - new_columns_names = ['a_plus_1'] + df.get_column_names() + new_columns_names = ['a_plus_1'] + df.column_names df = df.insert_column(new_column.rename('a_plus_1')) df = df.select(new_column_names) @@ -268,7 +268,8 @@ def rename_columns(self, mapping: Mapping[str, str]) -> DataFrame: """ ... - def get_column_names(self) -> list[str]: + @property + def column_names(self) -> list[str]: """ Get column names. diff --git a/spec/purpose_and_scope.md b/spec/purpose_and_scope.md index a48d7bdf..8f0d989d 100644 --- a/spec/purpose_and_scope.md +++ b/spec/purpose_and_scope.md @@ -291,7 +291,7 @@ def my_dataframe_agnostic_function(df): mask = df.get_column_by_name('species') != 'setosa' df = df.filter(mask) - for column_name in df.get_column_names(): + for column_name in df.column_names if column_name == 'species': continue new_column = df.get_column_by_name(column_name)