Create CSV file within Python using 'for' or 'if' conditions

Question

I have designed code as:

import csv
import numpy as np
data = [['Diameter', 'color', 'no']]
with open('samp1.csv', 'w') as f:
    writer = csv.writer(f, delimiter=',')
    for row in data:
        writer.writerow(row)
    for i in np.arange(20,30,0.2):
        writer.writerow(i)

f.close()

And I want to save numbers from 20-30 with an increment of 0.2 in the diameter column, but it's giving errors and not even saving the CSV file. Can someone suggest any solution? Even there are various ranges for other columns too, so I need the same method to work for that code.

CSV example

diameter    color     number
20          2         3
20          2.5       3
20          3         3
20          3.5       3
20.2        2         3
20.2        2.5       3
20.2        3         3
20.2        3.5       3
.
.
.
.
22          2         4
22          2.5       4
22          3         4
22          3.5       4
22.2        2         4
22.2        2.5       4
22.2        3         4
22.2        3.5       4

its just sample code from my program where I have lots of columns and for each column I need to save different kind of data like in first column numbers from 20-30 but with difference of .2 and in another column 2-5 with difference .5 each particular number needed to be matched with new one.. — pashaa
– pashaa, Commented Aug 8, 2018 at 9:13
@Vikas Damodar I have added sample o/p so can you guide me please. Thankyou — pashaa
– pashaa, Commented Aug 8, 2018 at 9:37

Peter Mortensen · Accepted Answer · 2018-09-04 05:09:41Z

1

I guess this can help you:

import csv

data = ['Diameter', 'color', 'no']
numb = [i for i in range(20,30,2)]
print(numb)
with open('samp1.csv', 'w') as f:
    writer = csv.writer(f, delimiter=',')
    writer.writerow(data)
    for i in numb:
            writer.writerow([i,i,i])

f.close()

edited Sep 4, 2018 at 5:09

Peter Mortensen

31.4k22 gold badges110 silver badges134 bronze badges

answered Aug 8, 2018 at 9:47

Rambod

2,7431 gold badge17 silver badges15 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

pashaa Over a year ago

Thankyou @RamG its working for me.And can I know if we want to save this range data in particular column is it possible?

Rambod Over a year ago

Happy to hear problem solved , please mark this as answer , yup u can do it , declare new array for them and put them instead of "i" in for i in numb: writer.writerow([i,i,i])

Jakub · Accepted Answer · 2018-08-08 09:11:43Z

1

What about using pandas library for creating it? Using it you can freely choose which part of DataFrame you want to save to csv.

import pandas as pd
d = {'col1': [1, 2], 'col2': [3, 4]}
df = pd.DataFrame(data=d)

df.to_csv('out.csv')

answered Aug 8, 2018 at 9:11

Jakub

112 bronze badges

1 Comment

pashaa Over a year ago

thankyou very much @Jakub but this will just display col1:[1,2] and col2:[3,4] I want to save data as col1: [range(20-30) increment by .2] and each number to be matched by another new value from col2:[range(2-5) increment by .5] can you guide me for this.

Collectives™ on Stack Overflow

Create CSV file within Python using 'for' or 'if' conditions

2 Answers 2

2 Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related