pandasでデータフレームに対しパラメータを追加・削除する

データフレームに列や行を追加したかったら concat() 、削除したかったら drop を使います。やってみましょう。

まず適当なデータフレームを作っておきます。

import pandas as pd
import numpy as np
df = pd.DataFrame(np.array([['a','b','c'], ['a','b','c'], ['a','b','c']]), columns = list("ABC"))

こんな感じになります。

   A  B  C
0  a  b  c
1  a  b  c
2  a  b  c

追加するようのデータも作っておきましょう

df_concat = pd.DataFrame(np.array(['d','d','d']), columns = list("D"))

Dの列を作りました

   D
0  d
1  d
2  d

さて、df にDの列を追加してみましょう。

df = pd.concat((df, df_concat),axis=1)

axis=1はこの場合列方向の削除を示します。 axis=0だと行方向です。期待通りDの列が追加されました。

   A  B  C  D
0  a  b  c  d
1  a  b  c  d
2  a  b  c  d

Dを削除してみましょう。

df = df.drop('D', axis=1)

削除もできました。

   A  B  C
0  a  b  c
1  a  b  c
2  a  b  c

こけこっこー