Как создать таблицу с разным количеством строк, используя prettytable модуль python

#python #prettytable

#python #prettytable

Вопрос:

У меня есть следующий код для создания таблицы с использованием PrettyTable:

 from prettytable import PrettyTable 

# columns names
columns = ["Characters", "FFF", "Job"] 

# lists
lista1 = ["Leanord", "Penny", "Howard", "Bernadette", "Sheldon", "Raj"]
lista2 = ["X", "X", "X", "X", "X", "X", "X"]
lista3 = ["B", "C", "A", "D", "A", "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!

Я был бы очень признателен за вашу помощь!!!

Ответ №1:

Сохраняйте все столбцы одинакового размера, заполняя более короткие столбцы пустыми строками.

 
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", "X"]
lista3 = ["B", "C", "A", "D", "A", "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     |     |     |                              
 ------------ ----- -----