pyhthonでデータフレーム の各要素に関数を適応する

Question

データフレームの各要素に引数と一緒に関数を適応したいです。
具体的には、データフレームの要素が小数点のfloat型になっているのですが、小数点第一で四捨五入したいです。

df.applymap(round)

とすると整数になってしまうので、引数1を渡したいのですが、どのようにしたら良いでしょうか？
他の方法でも構いません。
よろしくお願いします。

Answer 1

DataFrameクラスはround() Methodを持ってますのでapplymap() を使う必要はありません

http://pandas.pydata.org/pandas-docs/version/0.22/generated/pandas.DataFrame.round.html

df.round(1)

Answer 2

functools.partial()を利用して書けます。

>>> import numpy as np
>>> a = (np.arange(9)*0.333).reshape(3,3)
>>> import pandas as pd
>>> df = pd.DataFrame(a)
>>> a
array([[0.   , 0.333, 0.666],
       [0.999, 1.332, 1.665],
       [1.998, 2.331, 2.664]])
>>> df.applymap(round)
   0  1  2
0  0  0  1
1  1  1  2
2  2  2  3
>>> from functools import partial
>>> df.applymap(partial(round, ndigits=1))
     0    1    2
0  0.0  0.3  0.7
1  1.0  1.3  1.7
2  2.0  2.3  2.7

参考：
10.2. functools — 高階関数と呼び出し可能オブジェクトの操作 — Python 3.6.5 ドキュメント #functools.partial
Python で部分適用 - functools モジュールの partial 関数 | すぐに忘れる脳みそのためのメモ

Answer 3

lambda式を使う方法。

df.applymap(lambda x:round(x, 0))