將熊貓數據框導出爲表格圖像

Question

是否可以將熊貓數據框導出爲圖像文件？像df.to_png()或df.to_table().savefig('table.png')。

此刻我使用df.to_csv()導出數據幀。然後我在Excel中打開這個csv文件，使數據看起來很漂亮，然後將Excel表格複製/粘貼到Powerpoint中作爲圖像。我看到matplotlib有一個.table()方法，但我很難讓它與我的df一起工作。

我使用的DF有5列& 5行，每個「單元格」是一個數字。

在此先感謝。

Answer 1

如果你有pdflatex和ImageMagick的安裝，你可以導出數據幀到特克斯，使用pdflatex將其轉換爲PDF文件，然後使用ImageMagick的PDF轉換成PNG：

import pandas as pd 
import numpy as np 
import subprocess 

df = pd.DataFrame({'d': [1., 1., 1., 2., 2., 2.], 
        'c': np.tile(['a', 'b', 'c'], 2), 
        'v': np.arange(1., 7.)}) 
filename = 'out.tex' 
pdffile = 'out.pdf' 
outname = 'out.png' 

template = r'''\documentclass[preview]{{standalone}} 
\usepackage{{booktabs}} 
\begin{{document}} 
{} 
\end{{document}} 
''' 

with open(filename, 'wb') as f: 
    f.write(template.format(df.to_latex())) 

subprocess.call(['pdflatex', filename]) 
subprocess.call(['convert', '-density', '300', pdffile, '-quality', '90', outname]) 

如果安裝phantomjs和ImageMagick的，你可以 導出數據框爲HTML，然後使用phantomjs到HTML轉換成PNG和ImageMagick的裁剪結果：

import pandas as pd 
import numpy as np 
import subprocess 

df = pd.DataFrame({'d': [1., 1., 1., 2., 2., 2.], 
        'c': np.tile(['a', 'b', 'c'], 2), 
        'v': np.arange(1., 7.)}) 
filename = '/tmp/out.html' 
outname = '/tmp/out.png' 
cropname = '/tmp/cropped.png' 

with open(filename, 'wb') as f: 
    f.write(df.to_html()) 
rasterize = '/path/to/phantomjs/examples/rasterize.js' 
subprocess.call(['phantomjs', rasterize, filename, outname]) 
subprocess.call(['convert', outname, '-trim', cropname]) 

Answer 2

隨着一些額外的代碼，你甚至可以讓輸出看起來體面：

import pandas as pd 
import numpy as np 
import matplotlib.pyplot as plt 
import six 

df = pd.DataFrame() 
df['date'] = ['2016-04-01', '2016-04-02', '2016-04-03'] 
df['calories'] = [2200, 2100, 1500] 
df['sleep hours'] = [2200, 2100, 1500] 
df['gym'] = [True, False, False] 


def render_mpl_table(data, col_width=3.0, row_height=0.625, font_size=14, 
        header_color='#40466e', row_colors=['#f1f1f2', 'w'], edge_color='w', 
        bbox=[0, 0, 1, 1], header_columns=0, 
        ax=None, **kwargs): 
    if ax is None: 
     size = (np.array(data.shape[::-1]) + np.array([0, 1])) * np.array([col_width, row_height]) 
     fig, ax = plt.subplots(figsize=size) 
     ax.axis('off') 

    mpl_table = ax.table(cellText=data.values, bbox=bbox, colLabels=data.columns, **kwargs) 

    mpl_table.auto_set_font_size(False) 
    mpl_table.set_fontsize(font_size) 

    for k, cell in six.iteritems(mpl_table._cells): 
     cell.set_edgecolor(edge_color) 
     if k[0] == 0 or k[1] < header_columns: 
      cell.set_text_props(weight='bold', color='w') 
      cell.set_facecolor(header_color) 
     else: 
      cell.set_facecolor(row_colors[k[0]%len(row_colors) ]) 
    return ax 

render_mpl_table(df, header_columns=0, col_width=2.0)