pythonのfor文で意図しない値が入ってしまう

Question

pythonを用いてポアソン回帰をしております。問題はポアソン回帰ではなく、その結果を用いてグラフを描く時点で、単純にfor文を用いて配列に値を入れる際に、意図しない動作になってしまっており困っております。

import numpy as np
import seaborn as sns
import pandas as pd
import matplotlib.pyplot as plt
import statsmodels.api as sm
import scipy as sp

d = pd.read_csv('Beer.csv',encoding='SHIFT-JIS')
d.head()

X = d[['temperature']]
y = d[['beer_number']]
model = sm.GLM(y, sm.add_constant(X), family=sm.families.Poisson())
res=model.fit()

# 予測値の作成
x_axis = np.arange(0, 45)
y_seq = x_axis

for i in range(len(x_axis)):
    print("i")
    print(i)
    XX=sp.array([1,x_axis[i]])
    print(XX)
    zz=XX.dot(res.params)
    print(zz)
    
    kk=np.exp(zz)
    print("kk")
    print(kk)
    
    y_seq[i] = kk
    print("yy")
    print(y_seq[i])

【結果】
i
0
[1 0]
0.4476074167974242
kk
1.5645643533866962
yy
1
i
1
[1 1]
0.5236838580746012
kk
1.6882354281048548
yy
1
i
2
[1 2]
0.5997602993517781
kk
1.82168208967493
yy
1
【質問したいこと】
上記で変数y_seqという配列に、ポアソン回帰の結果のyの予測値を入れたいのですが、なぜか
意図しない整数値ばかりが入ってしまいます。不思議なことに、zzという値をネイピア数のべきにした値、kkというところまではprintすると想定通りの値が入っているのですが、それを単純にそのままy_seq[i]に入れた後に出力すると、そちらの結果は切り下げ？た様な整数値が入ってしまいます。何等かのバグでしょうか、、。ご教示のほど、宜しくお願い申し上げます。

なお、同じことをRで実装すると全く問題なく意図する結果となります。

Answer 1

整数型のnumpy配列に小数点数を代入すると整数型に変換されます。

>>> import numpy as np
>>> a = np.arange(3)
>>> a
array([0, 1, 2])
>>> a.dtype
dtype('int64')
>>> a[0] = 1.414
>>> a
array([1, 1, 2])

対策：最初からnumpy.float64の配列にする。

>>> a = np.arange(3, dtype=np.float64)
>>> a
array([0., 1., 2.])
>>> a[0] = 1.414
>>> a
array([1.414, 1.   , 2.   ])

Answer 2

済みません、上記ですが、以下２点により解決致しました。
①x_axis = np.linspace(0, 45,30)　とする。
②for文終了後に、改めてx_axis = np.linspace(0, 45,30)　とする。

②を行わない場合、なぜかx_axisの値もy_seqと同じになってしまい、
直線y=xのグラフになってしまいます（pythonのバグと思われます）。

Comments

[TaniguchiTakaki]

多分Pythonのバグではないです。
y_seq = x_axis
の動作を理解してないのではないでしょうか？

[hayataka2049]

バグではありません。
①はnumpyのたぶん明文化された仕様（リファレンスは拾っていないので、そういう記述がないかどうか探してみてください）。②は代入がコピーを作らないからで、Pythonの基本的な仕様です（参照：https://docs.python.org/ja/3/faq/programming.html#id17）