Отображение DataFrame с математическим символом в заголовках столбцов в Jupyter Notebook - PullRequest
1 голос
/ 29 апреля 2020

У меня есть DataFrame, который я хотел бы отобразить с греческими именами агрегатных функций.

df=pd.DataFrame(
      [["A",1,2],["A",3,4],["B",5,6],["B",7,8]], 
      columns=["AB","C", "N"]
)
df=df.groupby(df.AB).agg({
     "C":["count", "sum", "mean", "std"], 
     "N":["sum", "mean", "std"]
})

, который выглядит следующим образом:

Example DataFrame

Я хотел бы создать что-то похожее на это:

enter image description here

Я смог произвести:

enter image description here

С

import pandas as pd
import matplotlib.pyplot as plt

data = [[str(cell) for cell in row] for row in df.values]
columns = [
    r"Count", 
    r"C $\Sigma$", 
    r"C $\bar{x}$", 
    r"C $\sigma$",
    r"N $\Sigma$", 
    r"N $\bar{x}$", 
    r"N $\sigma$"]
rows = ["A", "B"]

the_table = plt.table(cellText=data,
                  rowLabels=rows,
                  colLabels=columns)

the_table.scale(4,4)
the_table.set_fontsize(24)
plt.tick_params(axis='x', which='both', bottom=False, top=False, labelbottom=False)
plt.tick_params(axis='y', which='both', right=False, left=False, labelleft=False)
for pos in ['right','top','bottom','left']:
    plt.gca().spines[pos].set_visible(False)    

Функция df.to_latex() выглядит так, как будто ее вполне может быть достаточно для моих целей, но она отображается в виде таблицы, которая не поддерживается jupyter.

Благодаря Elliot ниже, что-то вроде этого работает очень хорошо

substitutions = {
    "sum":"\u03a3",
    "mean":"\u03bc",
    "std":"\u03c3",
    True:"\u2705",
    False:"\u274c",
    "count":"N",
}

pretty = df.\
    rename(substitutions, axis=0).\
    rename(substitutions, axis=1)

и с:

%%HTML
<style type="text/css">
table.dataframe td, table.dataframe th {
    border: 1px  black solid !important;
  color: black !important;
}
th {
  text-align: center !important;
}
</style>

Может производить

enter image description here

1 Ответ

1 голос
/ 29 апреля 2020

Вы можете использовать символы Юникода для получения желаемых заголовков символов, не обращая внимания на to_latex().

Если вам нужны границы, вы можете использовать to_html с настраиваемыми параметрами для форматирования таблицы.

...