J'ai un fichier texte qui contient une matrice de N * M dimensions.
Par exemple, le fichier input.txt contient les éléments suivants:
0,0,0,0,0,0,0,0,0,0
0,0,0,0,0,0,0,0,0,0
0,0,0,0,0,0,0,0,0,0
0,0,0,0,0,0,0,0,0,0
0,0,0,0,0,0,0,0,0,0
0,0,0,0,0,0,0,0,0,0
0,0,2,1,0,2,0,0,0,0
0,0,2,1,1,2,2,0,0,1
0,0,1,2,2,1,1,0,0,2
1,0,1,1,1,2,1,0,2,1
J'ai besoin d'écrire un script python où je peux importer la matrice.
Mon script Python actuel est:
f = open ( 'input.txt' , 'r')
l = []
l = [ line.split() for line in f]
print l
la liste de sortie vient comme ça
[['0,0,0,0,0,0,0,0,0,0'], ['0,0,0,0,0,0,0,0,0,0'], ['0,0,0,0,0,0,0,0,0,0'],
['0,0,0,0,0,0,0,0,0,0'], ['0,0,0,0,0,0,0,0,0,0'], ['0,0,0,0,0,0,0,0,0,0'],
['0,0,2,1,0,2,0,0,0,0'], ['0,0,2,1,1,2,2,0,0,1'], ['0,0,1,2,2,1,1,0,0,2'],
['1,0,1,1,1,2,1,0,2,1']]
J'ai besoin de récupérer les valeurs sous forme int. Si j'essaie de taper cast, cela génère des erreurs.
Considérer
f = open ( 'input.txt' , 'r')
l = [[int(num) for num in line.split(',')] for line in f ]
print l
produit
[[0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 2, 1, 0, 2, 0, 0, 0, 0], [0, 0, 2, 1, 1, 2, 2, 0, 0, 1], [0, 0, 1, 2, 2, 1, 1, 0, 0, 2], [1, 0, 1, 1, 1, 2, 1, 0, 2, 1]]
Notez que vous devez séparer sur des virgules.
Si vous avez des lignes vides, changez
l = [[int(num) for num in line.split(',')] for line in f ]
à
l = [[int(num) for num in line.split(',')] for line in f if line.strip() != "" ]
Vous pouvez simplement utiliser numpy.loadtxt . Facile à utiliser. Vous pouvez également spécifier votre délimiteur, vos types de données, etc.
spécifiquement, tout ce que vous devez faire est ceci:
import numpy as np
input = np.loadtxt("input.txt", dtype='i', delimiter=',')
print(input)
Et le résultat serait:
[[0 0 0 0 0 0 0 0 0 0]
[0 0 0 0 0 0 0 0 0 0]
[0 0 0 0 0 0 0 0 0 0]
[0 0 0 0 0 0 0 0 0 0]
[0 0 0 0 0 0 0 0 0 0]
[0 0 0 0 0 0 0 0 0 0]
[0 0 2 1 0 2 0 0 0 0]
[0 0 2 1 1 2 2 0 0 1]
[0 0 1 2 2 1 1 0 0 2]
[1 0 1 1 1 2 1 0 2 1]]
Vous ne devriez pas écrire votre analyseur csv, considérez le module csv
lors de la lecture de tels fichiers et utilisez l'instruction with
pour fermer après avoir lu:
import csv
with open('input.txt') ad f:
data = [map(int, row) for row in csv.reader(f)]
Tu peux le faire:
fin = open('input.txt','r')
a=[]
for line in fin.readlines():
a.append( [ int (x) for x in line.split(',') ] )
Ce qui suit fait ce que vous voulez:
l = []
with open('input.txt', 'r') as f:
for line in f:
line = line.strip()
if len(line) > 0:
l.append(map(int, line.split(',')))
print l
Découvrez ce petit code d'une ligne pour lire la matrice,
matrix = [[input() for x in range(3)] for y in range(3)]
ce code lira la matrice d'ordre 3 * 3.
import numpy as np
f = open ( 'input.txt' , 'r')
l = []
l = np.array([ line.split() for line in f])
print (l)
type(l)
sortie:
[['0'] ['0'] ['0'] ['0,0,0,0,0,0,0,0,0,0,0'] ['0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0'] ['0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0']]
numpy.ndarray