Guardar resultados en el archivo csv con Python

Question

import csv 

with open('test.csv', 'rb') as f: 
    data = list(csv.reader(f)) 

import collections 
counter = collections.defaultdict(int) 
for row in data: 
    counter[row[1]] += 1 
for row in data: 
    if counter[row[1]] >= 4: 
     writer = csv.writer(open("test1.csv", "wb")) 
     writer.writerows(row) 

¡Estoy obteniendo resultados extraños! ¿Qué está mal con este código?

Answer 1

Uso csv.writer:

import csv 

with open('thefile.csv', 'rb') as f: 
    data = list(csv.reader(f)) 

import collections 
counter = collections.defaultdict(int) 
for row in data: 
    counter[row[0]] += 1 


writer = csv.writer(open("/path/to/my/csv/file", 'w')) 
for row in data: 
    if counter[row[0]] >= 4: 
     writer.writerow(row) 

Answer 2

Un ejemplo sencillo sería algo así como:

 
writer = csv.writer(open("filename.csv", "wb")) 
String[] entries = "first#second#third".split("#"); 
writer.writerows(entries) 
writer.close() 

Answer 3

Puede cerrar archivos no csv.writer objeto, que debe ser:

f = open(fileName, "wb") 
writer = csv.writer(f) 
String[] entries = "first*second*third".split("*"); 
writer.writerows(entries) 
f.close() 

Answer 4

Para su información, hay otras opciones que son más simples que el escritor — numpy, por ejemplo.

import numpy as np 
np.savetxt('data.csv', (col1_array, col2_array, col3_array), delimiter=',') 

(Esta respuesta hace 6 años más tarde, por el bien de la posteridad.)

Answer 5

Esto es cómo lo hago

import csv 
    file = open('???.csv', 'r') 
    read = csv.reader(file) 
    for column in read: 
      file = open('???.csv', 'r') 
      read = csv.reader(file) 
      file.close() 
      file = open('????.csv', 'a', newline='') 
      write = csv.writer(file, delimiter = ",") 
      write.writerow((,)) 
      file.close()