Deleting rows with Python in a CSV file

by

You are very close; currently you compare the row[2] with integer 0, make the comparison with the string "0". When you read the data from a file, it is a string and not an integer, so that is why your integer check fails currently:

row[2]!="0":

Also, you can use the with keyword to make the current code slightly more pythonic so that the lines in your code are reduced and you can omit the .close statements:

import csv
with open('first.csv', 'rb') as inp, open('first_edit.csv', 'wb') as out:
    writer = csv.writer(out)
    for row in csv.reader(inp):
        if row[2] != "0":
            writer.writerow(row)

Note that input is a Python builtin, so I’ve used another variable name instead.

Edit: The values in your csv file’s rows are comma and space separated; In a normal csv, they would be simply comma separated and a check against "0" would work, so you can either use strip(row[2]) != 0, or check against " 0".

The better solution would be to correct the csv format, but in case you want to persist with the current one, the following will work with your given csv file format:

$ cat test.py 
import csv
with open('first.csv', 'rb') as inp, open('first_edit.csv', 'wb') as out:
    writer = csv.writer(out)
    for row in csv.reader(inp):
        if row[2] != " 0":
            writer.writerow(row)
$ cat first.csv 
6.5, 5.4, 0, 320
6.5, 5.4, 1, 320
$ python test.py 
$ cat first_edit.csv 
6.5, 5.4, 1, 320

More Related Contents:

  1. How do I create a python code that distinguishes between the capital of a state and the state that the capital has?
  2. How do I read a large csv file with pandas?
  3. How to add pandas data to an existing csv file?
  4. How to ignore the first line of data when processing CSV data?
  5. Create a .csv file with values from a Python list
  6. Change specific value in CSV file via Python
  7. TypeError: a bytes-like object is required, not ‘str’ in python and CSV
  8. _csv.Error: field larger than field limit (131072)
  9. Remove file after Flask serves it
  10. General Unicode/UTF-8 support for csv files in Python 2.6
  11. Convert commas decimal separators to dots within a Dataframe
  12. Read data from CSV file and transform from string to correct data-type, including a list-of-integer column
  13. Inline CSV File Editing with Python
  14. ‘utf-8’ codec can’t decode byte 0xa0 in position 4276: invalid start byte
  15. What does ‘killed’ mean when processing a huge CSV with Python, which suddenly stops?
  16. Convert HTML into CSV
  17. How to *actually* read CSV data in TensorFlow?
  18. Loading UTF-8 file in Python 3 using numpy.genfromtxt
  19. Filter string data based on its string length
  20. Why is pandas.to_datetime slow for non standard time format such as ‘2014/12/31’
  21. UnicodeDecodeError: ‘utf-8’ codec can’t decode byte 0x96 in position 35: invalid start byte
  22. Simple CSV to XML Conversion – Python
  23. csv.writer writing each character of word in separate column/cell
  24. Pandas ParserError EOF character when reading multiple csv files to HDF5
  25. Python Pandas – Read csv file containing multiple tables
  26. Is there a way to speed up handling large CSVs and dataframes in python?
  27. Pandas read_csv dtype read all columns but few as string
  28. csv: writer.writerows() splitting my string inputs
  29. How can I stop Python’s csv.DictWriter.writerows from adding empty lines between rows in Windows?
  30. How to convert csv to json in python?

Leave a Comment