How do I save table from prettytable?

Question

So I am using the library 'prettytable' and am able to get a table of my data printed upon execution. However, I have no idea how to save it to a file. Ideally I'd like the table to be saved in a pdf so I could use it for my report.

Here is an example code from the prettytable webpage itself:

from prettytable import PrettyTable

x = PrettyTable(["City name", "Area", "Population", "Annual Rainfall"])
x.align["City name"] = "l" # Left align city names
x.padding_width = 1 # One space between column edges and contents (default)
x.add_row(["Adelaide",1295, 1158259, 600.5])
x.add_row(["Brisbane",5905, 1857594, 1146.4])
x.add_row(["Darwin", 112, 120900, 1714.7])
print x

Answer 1

Use get_string() method, docstring quote:

def get_string(self, **kwargs):

    """Return string representation of table in current state.
       ...
    """

Example:

from prettytable import PrettyTable

x = PrettyTable(["City name", "Area", "Population", "Annual Rainfall"])

...

data = x.get_string()

with open('test.txt', 'wb') as f:
    f.write(data)

This will create a test.txt file with your prettytable inside.

Hope that helps.

Answer 2

@Komal Jaswani You should change 'wb' mode to 'w' mode.

with open('test.txt', 'w') as f: f.write(data)