#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 | | |
------------ ----- -----