根据用户使用 Python 输入在表中添加或减去列/行

Question

我正在编辑这篇文章，因为我没有很好地解释我的问题，所以又来了。我正在使用制表库。我希望使用input() 根据给定的输入自动调整表中的列/行数。这是一个基本示例：

#!/bin/python3
from tabulate import tabulate
uid1 = input('UID > ')
uid2 = input('UID > ')
name1 = input('NAME > ')
name2 = input('NAME > ')
number1 = input('NUMBER > ')
number2 = input('NUMBER > ')
headers = ["UID", "NAME","NUMBER"]
table = [[uid1,name1,number1],[uid2,name2,number2]]
print(tabulate(table, headers, tablefmt="fancy_grid"))

╒═══════╤═════════╤══════════╕
│   UID │ NAME    │ NUMBER   │
╞═══════╪═════════╪══════════╡
│     0 │ SHAWN   │ 333-4444 │
├───────┼─────────┼──────────┤
│     1 │ MICHAEL │ 222-3333 │
╘═══════╧═════════╧══════════╛

但下次脚本运行时，列/表会更多：

╒═══════╤════════╤══════════╕
│   UID │ NAME   │ NUMBER   │
╞═══════╪════════╪══════════╡
│     0 │ JAMES  │ 444-5555 │
├───────┼────────┼──────────┤
│     1 │ ANDREW │ 666-3333 │
├───────┼────────┼──────────┤
│     2 │ SHAWN  │ 444-3333 │
╘═══════╧════════╧══════════╛

所以我试图找出一种方法来使用 for 循环或其他方式调整行和列的给定值，但我无法弄清楚。我想要类似的东西：

UIDs = input('Enter all UIDs $ ')
NAMES = input('Enter all names $ ')
NUMBERS = int(input('Enter all numbers $')

Answer 1

您可以使用while 和for 循环：

代码

---

from tabulate import tabulate

def adjust_columns_sizes(columns):
    """
    Adjusts the size of all the
    columns to be the same.
    
    If a column is smaller than
    the biggest, empty strings are added.
    
    Returns a list of lists of strings
    with fixed size.
    
    [['data1', 'data2', 'data3'], <-- column
     ['data']]                    <-- column
    
       ||
       ||
       \/
       
    [['data1', 'data2', 'data3'], <-- column
     ['data', '', '']]            <-- column
    """
    
    columns_sizes = []  # The size of each column
    for column in columns:
        columns_sizes.append(len(column))
    
    # Biggest column size is needed to adjust the
    # smaller columns to the same size.
    biggest_column_size = max(columns_sizes)
    
    # This loop adjust the smaller columns
    fixed_size_columns = []
    for i_column, column in enumerate(columns):  # Enumerate to get the index
        column_length = columns_sizes[i_column]
        if column_length < biggest_column_size:
            
            # Difference of lengths to know how many blank strings
            # to add.
            diff = biggest_column_size - len(column)
            
            # Using list comprehension, and adding of list, the blank
            # strings get added.
            column_with_new_size = column + ['' for _ in range(diff)]
            
            fixed_size_columns.append(column_with_new_size)
        else:
            # If column size is the same as the biggest, just add
            # to the list
            fixed_size_columns.append(column)

    return fixed_size_columns
        
    

def get_headers():
    """
    This function asks for all the headers
    that will be in the table.
    
    Returns a list of strings
    
    ['header', ...]
    """
    
    print('--HEADERS--')
    headers = []  # Headers stored in a list
    while True:
        new_h = input('New header value: ')
        if new_h == '':  # When no value is provided, while loop ends
            break
        headers.append(new_h)

    return headers


def get_body(headers):
    """
    This function asks for the data that will
    be in each column. 
    
    The number of columns is equivalent to the 
    number of headers or the lenght of the list 
    of headers.
    
    Each column will result in a list, but in order to be
    pased as an argument for tabulate, each list
    has to be a row, so we need to transpose
    the columns to rows using the zip function.
    
    Returns a list of tuples of strings. 
    
    [('data', ...), ('data', ...), ...]
    """

    print('--BODY--')
    body = []  # This list will contain the column lists
    for header in headers:
        
        # Message to show for each header
        i_msg = 'Enter all ' + header.lower() + '\'s separated by spaces: '
        
        # getting the data
        data_for_column = input(i_msg)  
        
        # getting a list splitting the data
        column = data_for_column.split()  
        
        body.append(column)
        
    # Adjusting the column sizes in order to zip to work properly
    adjusted_body = adjust_columns_sizes(body)
    
    # Transpose the columns to rows.
    # Using the * operator to pass every column in the list.
    # Using list() because zip returns an iterable object.
    transposed = list(zip(*adjusted_body))
   
    return transposed


if __name__ == '__main__':
    # Getting table headers
    headers = get_headers()
    
    # Getting the data en each column for each header
    # but transposed as rows
    table = get_body(headers)
    
    print(tabulate(table, headers, tablefmt="fancy_grid"))

结果

---

--HEADERS--
New header value: UID
New header value: NAME
New header value: NUMBER
New header value:
--BODY--
Enter all uid's separated by spaces: 0 1 2 3
Enter all name's separated by spaces: SHAWN MICHAEL JAMES ANDREW
Enter all number's separated by spaces: 333-444 222-333 444-555
╒═══════╤═════════╤══════════╕
│   UID │ NAME    │ NUMBER   │
╞═══════╪═════════╪══════════╡
│     0 │ SHAWN   │ 333-444  │
├───────┼─────────┼──────────┤
│     1 │ MICHAEL │ 222-333  │
├───────┼─────────┼──────────┤
│     2 │ JAMES   │ 444-555  │
├───────┼─────────┼──────────┤
│     3 │ ANDREW  │          │
╘═══════╧═════════╧══════════╛