Tengo un archivo que consta de dos columnas, es decir,
1 a
2 b
3 cDeseo leer este archivo en un diccionario de modo que la columna 1 sea la clave y la columna 2 sea el valor, es decir,
d = {1:'a', 2:'b', 3:'c'}El archivo es pequeño, por lo que la eficiencia no es un problema.
Respuestas:
d = {}
with open("file.txt") as f:
for line in f:
(key, val) = line.split()
d[int(key)] = valwithse usa aquí para manejar la limpieza del archivo. Cuando abandone el bloque (ya sea por el flujo de ejecución normal o por una excepción), el archivo se cerrará automáticamente. Puede leer más sobre los administradores de contexto en Python aquí: effbot.org/zone/python-with-statement.htm
for line in open("file.txt"):haga la limpieza de la misma manera. Y si f es un valor local, fse libera cuando se pierde el alcance. El único caso en el que esta declaración es útil es para una función larga (no es buena para la calidad) o si usa una variable global.
for line in open('file.txt')no no hacer la limpieza de la misma manera. No todas las implementaciones de Python son iguales. withgarantiza que el archivo se cerrará cuando se salga del bloque. Cuando la forlínea esté completa, close se puede llamar. CPythonlo hará, pero versiones como IronPythontienen recolectores de basura perezosos.
Esto dejará la clave como una cadena:
with open('infile.txt') as f:
d = dict(x.rstrip().split(None, 1) for x in f)dict([line.split() for line in f])es suficiente, en mi opinión.
dict([x.rstrip().split(None, 1) for x in f])lugar de dict(x.rstrip().split(None, 1) for x in f). Para aquellos que piensan lo mismo, la primera es una expresión generadora en lugar de una lista de comprensión como se explica aquí: python.org/dev/peps/pep-0289(PEP-289) . ¡Aprendí algo nuevo!
Si su versión de Python es 2.7+, también puede usar una comprensión de dictado como:
with open('infile.txt') as f:
{int(k): v for line in f for (k, v) in (line.strip().split(None, 1),)}def get_pair(line):
key, sep, value = line.strip().partition(" ")
return int(key), value
with open("file.txt") as fd:
d = dict(get_pair(line) for line in fd)partition? y withdeclaración?
partitiones más rápido y está creado exactamente para este propósito.
withes una forma sencilla de asegurarse de que lo sea.
strip, diría yo.
Por comprensión de diccionario
d = { line.split()[0] : line.split()[1] for line in open("file.txt") }O por pandas
import pandas as pd
d = pd.read_csv("file.txt", delimiter=" ", header = None).to_dict()[0]En mi humilde opinión, un poco más pitónico para usar generadores (probablemente necesite 2.7+ para esto):
with open('infile.txt') as fd:
pairs = (line.split(None) for line in fd)
res = {int(pair[0]):pair[1] for pair in pairs if len(pair) == 2 and pair[0].isdigit()}Esto también filtrará las líneas que no comiencen con un número entero o que no contengan exactamente dos elementos
import re
my_file = open('file.txt','r')
d = {}
for i in my_file:
g = re.search(r'(\d+)\s+(.*)', i) # glob line containing an int and a string
d[int(g.group(1))] = g.group(2)re? ¿seriamente?
split()no funciona casi en silencio si el formato de archivo no es sensato.
Si te encantan los one liners, prueba:
d=eval('{'+re.sub('\'[\s]*?\'','\':\'',re.sub(r'([^'+input('SEP: ')+',]+)','\''+r'\1'+'\'',open(input('FILE: ')).read().rstrip('\n').replace('\n',',')))+'}')Archivo de entrada = ruta al archivo, SEP = carácter separador de valor-clave
No es la forma más elegante o eficiente de hacerlo, pero no obstante es bastante interesante :)
Aquí hay otra opción ...
events = {}
for line in csv.reader(open(os.path.join(path, 'events.txt'), "rb")):
if line[0][0] == "#":
continue
events[line[0]] = line[1] if len(line) == 2 else line[1:]La mayoría de los métodos para almacenar un diccionario utilizan JSON, Pickle o lectura de líneas. Siempre que no esté editando el diccionario fuera de Python, este método simple debería ser suficiente incluso para diccionarios complejos. Aunque Pickle será mejor para diccionarios más grandes.
x = {1:'a', 2:'b', 3:'c'}
f = 'file.txt'
print(x, file=open(f,'w')) # file.txt >>> {1:'a', 2:'b', 3:'c'}
y = eval(open(f,'r').read())
print(x==y) # >>> True