python 2.7 script to add a column

by

Given that the question you posed doesn’t have enough information, I’ve formulated an answer that implements the rules you wrote about. Unfortunately, it’s hard to validate this because the expected output you provided doesn’t follow the rules (e.g. 133344444,3,106029,106961,12981,3_1category). I’ve also made the assumption that if an NA value is found, it should be treated as a zero integer value.

def compute_from_row(row):
    # Replace NA with value of 0
    row = ['0' if 'NA' in x else x for x in row]
    # Convert to integers ignoring the first column
    row[1:] = map(int, row[1:])
    numbers = row[2:5]
    # Implement the rules
    if row[1] == 3 and all(map(lambda x: x<=3000, numbers)):
      return '3_category'
    elif row[1] == 3 and all(map(lambda x: x<15000 and x>10000, numbers)):
      return '3_1category'
    elif row[1] == 2 and all(map(lambda x: x<=3000, numbers)):
      return '2_category'
    elif row[1] == 2 and all(map(lambda x: x<15000 and x>10000, numbers)):
      return '2_1category'
    elif row[1] == 1:
      return '1_category'



import csv


with open('1.csv') as k1, open('out.csv', 'w') as k2:
    reader = csv.reader(k1)
    writer = csv.writer(k2)

    headers = next(reader)
    headers.append('new_column')
    writer.writerow(headers)

    for row in reader:
        new_value = compute_from_row(row)
        row.append(new_value)
        writer.writerow(row)

More Related Contents:

  1. In Python, how do you convert a `datetime` object to seconds?
  2. Understanding recursion in Python
  3. Python Requests – No connection adapters
  4. Check if a process is running or not on Windows?
  5. Python handling socket.error: [Errno 104] Connection reset by peer
  6. Error: ‘int’ object is not subscriptable – Python
  7. Convert list to tuple in Python
  8. UnicodeDecodeError: ‘ascii’ codec can’t decode byte 0xe2 in position 13: ordinal not in range(128)
  9. Most pythonic way to interleave two strings
  10. Python : How to parse the Body from a raw email , given that raw email does not have a “Body” tag or anything
  11. How do I install the yaml package for Python?
  12. Search for a value in a nested dictionary python
  13. How can I do a batch insert into an Oracle database using Python?
  14. How to run different python versions in cmd [duplicate]
  15. Permission problems when creating a dir with os.makedirs in Python
  16. Unable to install sasl-0.1.3 python package on windows machine
  17. How to convert integer into date object python?
  18. Trying to use open(filename, ‘w’ ) gives IOError: [Errno 2] No such file or directory if directory doesn’t exist
  19. Using Properties in Python classes cause “maximum recursion depth exceeded” [duplicate]
  20. Screen Capture with OpenCV and Python-2.7
  21. Sound generation / synthesis with python?
  22. What are the differences between bool() and operator.truth()?
  23. how to open chrome in incognito mode from Python
  24. Matplotlib just giving error messages
  25. Is the right-hand side of an assignment always evaluated before the assignment?
  26. Module imports and __init__.py
  27. python_2_unicode_compatible error
  28. What does backward-slash b do in Python?
  29. How to filter a DataFrame column of lists for those that contain a certain item
  30. How to display charts in Spyder

Leave a Comment