问题描述
我有以下代码使用 prettytable 生成表格:
from prettytable import prettytable
# columns names
columns = ["Characters","FFF","Job"]
# lists
lista1 = ["Leanord","Penny","Howard","Bernadette","Sheldon","Raj"]
lista2 = ["X","X","X"]
lista3 = ["B","C","A","D","B","B"]
# init table
myTable = prettytable()
# Add data
myTable.add_column(columns[0],[item for item in lista1])
myTable.add_column(columns[1],[item for item in lista2])
myTable.add_column(columns[2],[item for item in lista3])
print(myTable)
这是输出:
但是,我真正想要的是创建一个具有不同列大小的表(我的列表具有不同的长度)。换句话说,这是我想要的输出:
我尝试使用上面的代码,但是,我收到以下错误:异常:列长度 7 与行数 6 不匹配!
非常感谢您的帮助!!!
解决方法
通过用空字符串填充较短的列来保持所有列的大小相同。
from prettytable import PrettyTable
def column_pad(*columns):
max_len = max([len(c) for c in columns])
for c in columns:
c.extend(['']*(max_len-len(c)))
# columns names
columns = ["Characters","FFF","Job"]
# lists
lista1 = ["Leonard","Penny","Howard","Bernadette","Sheldon","Raj","Amy"]
lista2 = ["X","X","X"]
lista3 = ["B","C","A","D","B"]
column_pad(lista1,lista2,lista3)
# init table
myTable = PrettyTable()
# Add data
myTable.add_column(columns[0],lista1)
myTable.add_column(columns[1],lista2)
myTable.add_column(columns[2],lista3)
print(myTable)
+------------+-----+-----+
| Characters | FFF | Job |
+------------+-----+-----+
| Leonard | X | B |
| Penny | X | C |
| Howard | X | A |
| Bernadette | X | D |
| Sheldon | | A |
| Raj | | B |
| Amy | | |
+------------+-----+-----+