table formatting, print out row names

Question

I have a dictionary I want to format into a table:

band3 = \
{'channel1': [10564, 2112, 1922],
 'channel10': [10787, 2157, 1967],
 'channel11': [10812, 2162, 1972],
 'channel12': [10837, 2167, 1977],
 'channel2': [10589, 2117, 1927],
 'channel3': [10612, 2122, 1932],
 'channel4': [10637, 2127, 1937],
 'channel5': [10662, 2132, 1942],
 'channel6': [10687, 2137, 1947],
 'channel7': [10712, 2142, 1952],
 'channel8': [10737, 2147, 1957],
 'channel9': [10762, 2152, 1962]}

I do so like this:

table = [[], [], [], []]
# can't just sort the channel names because 'channel11' < 'channel2'
channel_numbers = []
for channel_name in band3.keys():
    if channel_name.startswith('channel'):
        channel_number = int(channel_name[7:])
        channel_numbers.append(channel_number)
    else:
        raise ValueError("channel name doesn't follow pattern")
channel_numbers.sort()

for channel_number in channel_numbers:
    channel_data = band2['channel%d' % channel_number]
    column =[
              'Channel %d' % channel_number,
               str(channel_data[0]),
               '%s/%s' % (channel_data[1], channel_data[2]),
               str(channel_data[3])
            ]
    cell_widths = map(len, column) #9 5 2 9
    column_widths = max(cell_widths) # 9 or 10
    for i in range(len(cell_widths)): #4
        cell = column[i]
        padded_cell = cell + ' '*(column_widths-len(cell))
        table[i].append(padded_cell)
for line in table:
    print('  '.join(line))

This gives:

Channel 1  Channel 2  Channel 3  Channel 4  Channel 5  Channel 6  Channel 7  Channel 8  Channel 9  Channel 10  Channel 11  Channel 12
10564      10589      10612      10637      10662      10687      10712      10737      10762      10787       10812       10837     
2112/1922  2117/1927  2122/1932  2127/1937  2132/1942  2137/1947  2142/1952  2147/1957  2152/1962  2157/1967   2162/1972   2167/1977 
20         0          0          26         32         0          26         0          0          0           0           15       

However now I would like to name the rows:

       Channel 1  Channel 2  Channel 3  Channel 4  Channel 5  Channel 6  Channel 7  Channel 8  Channel 9  Channel 10  Channel 11  Channel 12
UARFCN 10564      10589      10612      10637      10662      10687      10712      10737      10762      10787       10812       10837     
DL/UL  2112/1922  2117/1927  2122/1932  2127/1937  2132/1942  2137/1947  2142/1952  2147/1957  2152/1962  2157/1967   2162/1972   2167/1977 
RSSI   20         0          0          26         32         0          26         0          0          0           0           15        

This is fairly easy, I change the print loop to this:

print "      ",
print('  '.join(table[0])) 
print "UARFCN",
print('  '.join(table[1])) 
print "DL/UL ",
print('  '.join(table[2])) 
print "RSSI  ",
print('  '.join(table[3])) 

However I'd like to know some nicer ways to print these column names. I could get verbose with padding calculations as above, but I was wondering what would be a nice clean, simple way.

EDIT: format attempt:

print('{0:6s}  {1}'.format("", ' '.join(table[0])))   
print('{0:2s}  {1}'.format("UARFCN", ' '.join(table[1])))   
print('{0:6s}  {1}'.format("DL/UL", ' '.join(table[2])))   
print('{0:6s}  {1}'.format("RSSI", ' '.join(table[3]))) 

EDIT: another way

print('{0} {1}'.format("".ljust(6), ' '.join(table[0])))   
print('{0} {1}'.format("UARFCN".ljust(6), ' '.join(table[1])))   
print('{0} {1}'.format("DL/UL".ljust(6), ' '.join(table[2])))   
print('{0} {1}'.format("RSSI".ljust(6), ' '.join(table[3])))  

Improvement suggestions?

Answer 1

As with so many python tasks, more elegant code can be achieved by importing magical modules. In this case, the module is called prettytable.

Here is some working code:

from prettytable import PrettyTable

band3 = \
{'channel1': [10564, 2112, 1922],
 'channel10': [10787, 2157, 1967],
 'channel11': [10812, 2162, 1972],
 'channel12': [10837, 2167, 1977],
 'channel2': [10589, 2117, 1927],
 'channel3': [10612, 2122, 1932],
 'channel4': [10637, 2127, 1937],
 'channel5': [10662, 2132, 1942],
 'channel6': [10687, 2137, 1947],
 'channel7': [10712, 2142, 1952],
 'channel8': [10737, 2147, 1957],
 'channel9': [10762, 2152, 1962]}

band3_new = {}
for key in band3.keys():
    band3_new[ int( key.split('channel')[1] ) ] = band3[key]

x = PrettyTable()
x.add_column("", ["UARFCN", "DL/UL"])

for channel in band3_new.keys():
    x.add_column("channel " + str(channel), [band3_new[channel][0], str(band3_new[channel][1]) + "/" + str(band3_new[channel][2]) ] )

print x

And its output:

+--------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+------------+------------+------------+
|        | channel 1 | channel 2 | channel 3 | channel 4 | channel 5 | channel 6 | channel 7 | channel 8 | channel 9 | channel 10 | channel 11 | channel 12 |
+--------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+------------+------------+------------+
| UARFCN |   10564   |   10589   |   10612   |   10637   |   10662   |   10687   |   10712   |   10737   |   10762   |   10787    |   10812    |   10837    |
| DL/UL  | 2112/1922 | 2117/1927 | 2122/1932 | 2127/1937 | 2132/1942 | 2137/1947 | 2142/1952 | 2147/1957 | 2152/1962 | 2157/1967  | 2162/1972  | 2167/1977  |
+--------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+-----------+------------+------------+------------+

I couldn't figure out where you get the RSSI row values from, though.

Answer 2

I guess this will help in the printing if you don't mind the order..:

band3 = \
{'channel1': [10564, 2112, 1922],
...
 'channel9': [10762, 2152, 1962]}

types = ['','UARFCN','DL/UL','RSSI']

# ASSUME THE DICTIONARY VALUES ARE EQUAL IN LENGTH
li1 = [types[0]]+band3.keys() # the stuff on the side
li2 = zip(types[1:],*band3.values()) # the main data
li3 = zip(li1,*li2)
for i,x in enumerate(li3):
    x = map(str, x)
    li3[i] = [a.ljust(b) for b in (max(map(len,x)),) for a in x]

final = zip(*li3)
for i in final:
    print ' '.join(i)

Output:

       channel3 channel2 channel1 channel7 channel6 channel5 channel4 channel9 channel8 channel12 channel11 channel10
UARFCN 10612    10589    10564    10712    10687    10662    10637    10762    10737    10837     10812     10787    
DL/UL  2122     2117     2112     2142     2137     2132     2127     2152     2147     2167      2162      2157     
RSSI   1932     1927     1922     1952     1947     1942     1937     1962     1957     1977      1972      1967