pythonでgroupbyを利用した数値計算結果をデータフレームに格納したところ1列に全部入れられてしまった

Question

前提

データ解析をしています。円形の等高線から場所ごとに線積分して場所ごとの線積分値をプロットして投影図を作りたいと考えています。自分はx軸とy軸を等分割して重複ありの組み合わせを作り出すことにより、円形等高線を形成するしきに代入した値をあとで合計すれば可能ではと考えプログラミングしています。しかし合計するという部分で詰まってしまいました。また併せてで同じ値を含んだ重複あり順列の作り方がわかりません。現在では同じ値同士の組み合わせを除外してしまっています。

実現したいこと

データフレーム内の1列目の値でグループ分けして、分けたグループごとで3列目の値の合計を出して、再度データフレームに格納したい。

発生している問題・エラーメッセージ

groupbyを試したところ値的には求めているものが得られたと思うのですが、本来100行×2列になると思うのですが、100行×1列で1列に二つの値が格納されてしまいました。

該当のソースコード

import numpy as np
import matplotlib.pyplot as plt
from scipy.special import jv
import pandas as pd
import itertools 

fig = plt.figure()
ax1 = fig.add_subplot(111)
x = np.linspace(0, 4*2^(1+0), 40) #等間隔で値を０～4まで20個作成 lの値はここで変える
y = np.linspace(0, 2*np.pi, 40) #等間隔で値を０～360まで20個作成
x, y = np.meshgrid(x, y)  #ｘ,ｙからメッシュグリッドを作成
def fz(x,y):            #高さデータzを作る関数fz定義
    return jv(0,x)*np.cos(0*y)    #ここで基底関数決める→jv(m,r)*cos or sin(mθ)
z = fz(x,y)

x1 = x*np.cos(y)
y1= x*np.sin(y)

ax = plt.contour(x1,y1,z,colors='black')    #等高線表示
ax.clabel(fmt='%1.1f', fontsize=16)       #等高線の値を表示
ax = plt.contourf(x1,y1,z,cmap='rainbow')   #等高線レベルに応じて色を塗る
ax = plt.colorbar(label="contour level")  #カラーバー表示


plt.show()

#x=rcosθ
#y=rsinθ

start = -8
stop = 8    #半径を入れる
num =100     #分割数
lst = np.linspace(start, stop, num)
perm = itertools.permutations(lst,2)
df = pd.DataFrame(perm)
df.columns=['x','y']
x_prj = df["x"]
y_prj = df["y"]
df["z"] = jv(0,np.sqrt(np.square(x_prj)+np.square(y_prj)))*np.cos(0*np.arctan(y_prj/x_prj))
z_prj = df["z"]
lst
df

df_gr = df.groupby('x').agg({'z': np.sum})
print(df_gr)

試したこと

最初はfor文で最小値から足し合わせようと考えましたが、数が合わなくなることに気づきましたのでgroupbyを利用しました。おそらく値的にはあっています。あとは1列に二つ格納されてカラムのzがどこかに行ってしまっています。以下の状態です。私のzはいったいどうしてこんなところに逃げ出してしまったのでしょうか？
z
x
-8.000000 -5.123848
-7.838384 -2.824273
-7.676768 -0.502862
-7.515152 1.781209
-7.353535 3.970557
... ...
7.353535 3.970557
7.515152 1.781209
7.676768 -0.502862
7.838384 -2.824273
8.000000 -5.123848

[100 rows x 1 columns]

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[melian]

df_gr = df.groupby('x').agg({'z': np.sum}) の後で、
print(df_gr.index)
print(df_gr.columns)

を実行してみると分かるかと思いますが、df.groupby('x') が df_gr のインデックスになっていて、z 列が df_gr のカラムになっています。

[NMKN]

melian様　確かにインデックスになってますね.... 引数にas_indexを追加で解決しました。ありがとうございます。

Answer 1

引数にas_index 追加で解決。xがインデックスになっていたようです。

df_gr = df.groupby(['x'],as_index=False).agg({'z': np.sum})
print(df_gr)