如何使用prettytable模块python创建具有不同行数的表

问题描述

我有以下代码使用 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)

这是输出

enter image description here

但是,我真正想要的是创建一个具有不同列大小的表(我的列表具有不同的长度)。换句话说,这是我想要的输出

enter image description here

我尝试使用上面的代码,但是,我收到以下错误异常:列长度 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     |     |     |                              
+------------+-----+-----+