listからカーブフィッティングをしたい

Question

データのファイルからカーブフィッティングしたいです．

y = np.poly1d(np.polyfit(x, y, 1))(x)

これでxを元とした整形したyの値を取れます．x,yはどちらも一次元配列です．

l =['0.0\t1.0\t-10.569\t1.835\n', '0.0\t2.0\t33.977\t-4.618\n', '12.0\t1.0\t-12.138\t1.632\n',...

のような要素が[a,b,c,d]となっています．
ここでbの要素のそれぞれのc,dから新しいdを上記の計算で求めたいです．

やりたいこと
・bの値をsorted(l, key=lambda x:x[1])で並び替える
・並び替えたlをbの要素ごとに分ける
・分けた要素ごとに上記のカーブフィッティングの式を当てはめる

自分でやってみたのですがうまくいきませんでした．
並び替えて要素に分け,配列を作ることが難しいです．
教えていただきたいです．お願いします．

Answer 1

並び替えて要素に分け,配列を作ることが難しいです、の部分はこれぐらいでしょうか。

>>> print(l)
['10.0\t6.0\t-12.136\t-2.499\n', '9.0\t5.0\t4.880\t-1.051\n', '3.0\t5.0\t6.072\t-0.630\n', '7.0\t3.0\t25.789\t0.801\n', '6.0\t3.0\t-13.696\t2.702\n', '4.0\t1.0\t-7.623\t-3.802\n', '5.0\t1.0\t15.421\t-4.963\n', '10.0\t2.0\t-13.247\t-2.756\n', '1.0\t3.0\t-0.897\t-3.588\n', '6.0\t3.0\t-1.648\t2.958\n', '6.0\t1.0\t-15.427\t-3.081\n', '7.0\t3.0\t19.761\t-1.834\n', '7.0\t5.0\t1.221\t2.980\n', '7.0\t6.0\t-0.506\t-0.423\n', '10.0\t4.0\t26.638\t-2.038\n', '5.0\t6.0\t-10.222\t-3.705\n', '6.0\t3.0\t26.416\t3.684\n', '5.0\t1.0\t11.384\t3.758\n', '4.0\t6.0\t7.098\t-0.432\n', '5.0\t2.0\t17.753\t-3.231\n']
>>>
>>> d = {}
>>> for x in l:
...     f = float(x.split('\t')[1])
...     if f in d:
...         d[f].append(x)
...     else:
...         d[f] = [x]
...
>>> sl = sorted(list(d.items()), key=lambda x: x[0])
>>>
>>> for i in sl:
...     print(i[0], ':')
...     print(i[1], '\n')
...
1.0 :
['4.0\t1.0\t-7.623\t-3.802\n', '5.0\t1.0\t15.421\t-4.963\n', '6.0\t1.0\t-15.427\t-3.081\n', '5.0\t1.0\t11.384\t3.758\n']

2.0 :
['10.0\t2.0\t-13.247\t-2.756\n', '5.0\t2.0\t17.753\t-3.231\n']

3.0 :
['7.0\t3.0\t25.789\t0.801\n', '6.0\t3.0\t-13.696\t2.702\n', '1.0\t3.0\t-0.897\t-3.588\n', '6.0\t3.0\t-1.648\t2.958\n', '7.0\t3.0\t19.761\t-1.834\n', '6.0\t3.0\t26.416\t3.684\n']

4.0 :
['10.0\t4.0\t26.638\t-2.038\n']

5.0 :
['9.0\t5.0\t4.880\t-1.051\n', '3.0\t5.0\t6.072\t-0.630\n', '7.0\t5.0\t1.221\t2.980\n']

6.0 :
['10.0\t6.0\t-12.136\t-2.499\n', '7.0\t6.0\t-0.506\t-0.423\n', '5.0\t6.0\t-10.222\t-3.705\n', '4.0\t6.0\t7.098\t-0.432\n']

Comments

[ikayakioishii]

手助けありがとうございます．
あとは自分でやってみます．

[ppaul]

がんばってください。