python - para - seleccionar columnas de un dataframe pandas

Python-Imprimir la lista de cadenas CSV en columnas alineadas

Question

He escrito un fragmento de código que es totalmente compatible con Python 2 y Python 3 . El fragmento que escribí analiza datos y construye el resultado como una lista de cadenas de CSV .

El script proporciona una opción para:

• escribir los datos en un CSV file , o
• mostrarlo a la stdout .

Si bien podría iterar fácilmente a través de la lista y reemplazarla , con /t cuando se muestre a stdout (segunda opción de viñeta), los ítems tienen una longitud arbitraria, por lo que no se alinee en un formato agradable debido a las variaciones en las pestañas.

He investigado un poco, y creo que las opciones de formato de cadena podrían lograr lo que busco. Dicho esto, parece que no puedo encontrar un ejemplo que me ayude a obtener la sintaxis correcta.

Preferiría no usar una biblioteca externa . Soy consciente de que hay muchas opciones disponibles si tomé esa ruta, pero quiero que el script sea lo más compatible y simple posible.

Aquí hay un ejemplo:

value1,somevalue2,value3,reallylongvalue4,value5,superlongvalue6
value1,value2,reallylongvalue3,value4,value5,somevalue6

¿Puedes ayudarme por favor? Cualquier sugerencia será muy apreciada.

Answer 1

def printCsvStringListAsTable(csvStrings):

    # convert to list of lists
    csvStrings = map(lambda x: x.split('',''), csvStrings)

    # get max column widths for printing
    widths = []
    for idx in range(len(csvStrings[0])):
        columns = map(lambda x: x[idx], csvStrings)
        widths.append(
            len(
                max(columns, key = len)
            )
        )

    # print the csv strings
    for row in csvStrings:
        cells = []
        for idx, col in enumerate(row):
            format = ''%-'' + str(widths[idx]) + "s"
            cells.append(format % (col))
        print '' |''.join(cells)


if __name__ == ''__main__'':
    printCsvStringListAsTable([
        ''col1,col2,col3,col4'',
        ''val1,val2,val3,val4'',
        ''abadfafdm,afdafag,aadfag,aadfaf'',
    ])

Salida:

col1      |col2    |col3   |col4
val1      |val2    |val3   |val4
abadfafdm |afdafag |aadfag |aadfaf

La respuesta de Alex Hall es definitivamente mejor y una forma concisa del mismo código que he escrito.

Answer 2

import csv
from StringIO import StringIO

rows = list(csv.reader(StringIO(
    ''''''value1,somevalue2,value3,reallylongvalue4,value5,superlongvalue6
value1,value2,reallylongvalue3,value4,value5,somevalue6'''''')))

widths = [max(len(row[i]) for row in rows) for i in range(len(rows[0]))]

for row in rows:
    print('' | ''.join(cell.ljust(width) for cell, width in zip(row, widths)))

Salida:

value1 | somevalue2 | value3           | reallylongvalue4 | value5 | superlongvalue6
value1 | value2     | reallylongvalue3 | value4           | value5 | somevalue6