[PYTHON] Praktische Funktion zum Hinzufügen von Spalten an einer beliebigen Stelle in Pandas DataFrame
Wenn Sie irgendwo im Pandas DataFrame eine Spalte hinzufügen möchten, verwenden Sie pandas.DataFrame.insert. Es kann erreicht werden, indem es verwendet wird, aber es gibt einige Beschwerden.
- Nicht unveränderlich (DataFrame ist direkt umgeschrieben / destruktive Methode / keine Inplace-Option)
- Der hinzuzufügende Ort muss durch den numerischen Wert des Index angegeben werden
- Schwer zu verstehen
- Es ist schwierig, etwas wie "Nach Spalte1 hinzufügen" anzugeben. --Name wird ignoriert, auch wenn Sie Series eingeben. Sie müssen ihn daher separat angeben. --Kann DataFrame nicht eingeben
Ich habe eine praktische Funktion geschrieben, um diese Probleme zu lösen.
Implementierung
from typing import Union, Optional
import pandas as pd
def insert_columns(
df: pd.DataFrame,
data: Union[pd.Series, pd.DataFrame],
*,
before: Optional[str] = None,
after: Optional[str] = None,
allow_duplicates: bool = False,
inplace: bool = False,
) -> pd.DataFrame:
if not inplace:
df = df.copy()
if not (after is None) ^ (before is None):
raise ValueError('Specify only "before" or "after"')
if before:
loc = df.columns.get_loc(before)
else:
loc = df.columns.get_loc(after) + 1
if type(data) is pd.Series:
df.insert(loc, data.name, data, allow_duplicates)
elif type(data) is pd.DataFrame:
for column in data.columns[::-1]:
df.insert(loc, column, data[column], allow_duplicates)
return df
- Geben Sie den Spaltennamen für "vor" oder "nach" an
- Standardmäßig wird der eingegebene DataFrame nicht direkt neu geschrieben.
- Schreiben Sie direkt neu, indem Sie "inplace = True" angeben
Wie benutzt man
Verwenden Sie den Iris-Datensatz von sklearn als Beispieldaten.
from sklearn import datasets
iris = datasets.load_iris()
df = pd.DataFrame(iris.data, columns=iris.feature_names)
target = pd.Series(iris.target_names[iris.target], name='target')
df.head()
target.head()
Versuchen Sie, das Ziel nach "Kelchblattbreite (cm)" in df hinzuzufügen.
insert_columns(df, target, after='sepal width (cm)')
In diesem Beispiel wird Series hinzugefügt, es kann jedoch auch DataFrame angegeben werden.
Recommended Posts