python - pandas DataFrame から LaTeX テーブルをエクスポートする

Question

データ フレーム (またはその一部) を LaTeX にエクスポートする簡単な方法はありますか?

Google で検索したところ、asciitables を使用したソリューションしか見つかりませんでした。

Answer 1

DataFrame にはto_latex( pandas docsを参照) メソッドがあります。

>>> df = pd.DataFrame(np.random.random((5, 5)))
>>> df  
          0         1         2         3         4
0  0.886864  0.518538  0.359964  0.167291  0.940414
1  0.834130  0.022920  0.265131  0.059002  0.530584
2  0.648019  0.953043  0.263551  0.595798  0.153969
3  0.207003  0.015721  0.931170  0.045044  0.432870
4  0.039886  0.898780  0.728195  0.112069  0.468485

>>> print(df.to_latex())
\begin{tabular}{|l|c|c|c|c|c|c|}
\hline
{} &         0 &         1 &         2 &         3 &         4 \\
\hline
0 &  0.886864 &  0.518538 &  0.359964 &  0.167291 &  0.940414 \\
1 &  0.834130 &  0.022920 &  0.265131 &  0.059002 &  0.530584 \\
2 &  0.648019 &  0.953043 &  0.263551 &  0.595798 &  0.153969 \\
3 &  0.207003 &  0.015721 &  0.931170 &  0.045044 &  0.432870 \\
4 &  0.039886 &  0.898780 &  0.728195 &  0.112069 &  0.468485 \\
\hline
\end{tabular}

これを tex ファイルに書き込むだけです。

デフォルトでは、ラテックスはこれを次のようにレンダリングします。

注: to_latex( pandas docsを参照) メソッドには、いくつかの構成オプションがあります。

Answer 2

テキストファイルに書き込むだけです。それは魔法ではありません:

import pandas as pd
df = pd.DataFrame({"a":range(10), "b":range(10,20)})
with open("my_table.tex", "w") as f:
    f.write("\\begin{tabular}{" + " | ".join(["c"] * len(df.columns)) + "}\n")
    for i, row in df.iterrows():
        f.write(" & ".join([str(x) for x in row.values]) + " \\\\\n")
    f.write("\\end{tabular}")