関数を適用させて、DataFrameの要素にndarrayを格納したい

Question

実現したいこと

下記のprint dfで出力されているような、DataFrameにndarrayのカラムを作りたいです。

import pandas as pd
import numpy as np

nd = np.asarray([[1,2,3],[4,5,6]])
df = pd.DataFrame({'hoge': [0,1,2],
                   'ndarray': [nd,nd,nd]},
                  index=['a','b','c'])
print df
#    hoge                 ndarray
# a     0  [[1, 2, 3], [4, 5, 6]]
# b     1  [[1, 2, 3], [4, 5, 6]]
# c     2  [[1, 2, 3], [4, 5, 6]]

本題

上のやり方だとうまくいくのですが、
DataFrameに関数を適用させて作りたいというのが本題です。

そこで、関数をapplyで適用させて作ろうとしたのですが、
１次元にしろとのエラーが出てしまいました。
改善、代替案が浮かばない状況です。

def input_ndarray(df):
    return np.asarray([[1,2,3],[4,5,6]])

df = pd.DataFrame({'hoge': [0,1,2]},
                  index=['a','b','c'])
df['ndarray'] = df.apply(input_ndarray, axis=1)

# Error

Exception                                 Traceback (most recent call last)
<ipython-______> in <module>()
      4 df = pd.DataFrame({'hoge': [0,1,2]},
      5                   index=['a','b','c'])
----> 6 df['ndarray'] = df.apply(input_ndarray, axis=1)

...

Exception: Data must be 1-dimensional

質問

DataFrameに関数を適用させて、要素にndarrayが入った列の作り方を、
どなたかご教授いただけないでしょうか？

Answer 1

単純に以下でよいかと思います。

import pandas as pd
import numpy as np

nd = np.asarray([[1,2,3],[4,5,6]])
df = pd.DataFrame({'hoge': [0,1,2]},
                  index=['a','b','c'])

df['ndarray'] = df.apply(lambda _:nd, axis=1)
print df
"""
   hoge                 ndarray
a     0  [[1, 2, 3], [4, 5, 6]]
b     1  [[1, 2, 3], [4, 5, 6]]
c     2  [[1, 2, 3], [4, 5, 6]]
"""

Comments

[o-range]

ご回答ありがとうございます！
結論から言うと、原因は開発環境にあるようでした。
考えが及ばずお騒がせしました。

ご教授いただいた方法でも、Exception: Data must be 1-dimensional のエラーを返されましたが、
別環境のPython 2.7で実行してみるとすんなりと実行してくれました。