assign the outputs in a loop to a single list or array

Question

I am still finding my way in Python so this question may be very basic: I am trying to generate some cosine waves in python and then do a Fourier transform on them. I am wrote the following code:

from pylab import *
from numpy import *
x = linspace(-5,5,100)
t = linspace(0,2000,2001)
w1=cos(2*pi*3*t/2001)
w2=cos(2*pi*10*t/2001)
fourier=0
long(fourier)

Trace = w1+w2
for i in range(1,125,1):
    fourier[i] = sum(cos(2*pi*i*t/2001)*Trace)

print fourier
f=linspace(1,125,125)
plot(f,fourier)
show()

so instead of calculating the values frequency by frequency I want to make a loop that takes each frequency from 1 to 125 and store the outputs in one list so that I can plot them.

Python does not like this and gives a message saying:

fourier[i] = sum(cos(2*pi*i*t/2001)*Trace)
TypeError: 'int' object does not support item assignment

Any idea why?

you need range(1,126) to match the the dimensions of f for plotting — Padraic Cunningham
– Padraic Cunningham, Commented Dec 25, 2014 at 12:01

Padraic Cunningham · Accepted Answer · 2014-12-25 12:09:19Z

3

may be you want:

fourier = []

you decleared it as integer, you need to declear as list.

your fourior type is integer and you need to make it as list so it can support index assignment

also remove long(fourier) from your code

for i in range(1,126):
    fourier[i] = sum(cos(2*pi*i*t/2001)*Trace)

this will give you index out of index error

correct would be:

for i in range(1,126):    # you dont need last option as 1, because range step is one by default
    fourier.append(sum(cos(2*pi*i*t/2001)*Trace))
 OR
    fourier.insert(i,sum(cos(2*pi*i*t/2001)*Trace))

its better to use list comprehension and you need to set range to 126 to match the dimensions of f for plotting:

fourier = [sum(cos(2*pi*i*t/2001)*Trace) for i in range(1,126)]

edited Dec 25, 2014 at 12:09

Padraic Cunningham

181k30 gold badges264 silver badges327 bronze badges

answered Dec 25, 2014 at 11:52

Hackaholic

19.8k6 gold badges59 silver badges77 bronze badges

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

5 Comments

jamylak Over a year ago

both of these answers will still produce an error when trying to access fourier[i] on a list with No items

Hackaholic Over a year ago

@PadraicCunningham yes mentioned it will give list out of index error :)

Padraic Cunningham Over a year ago

better to just use a list comp and the range needs to be 126

Hackaholic Over a year ago

@PadraicCunningham Happy Christmas :)

Padraic Cunningham Over a year ago

happy christmas to you too :)

nitishagar · Accepted Answer · 2014-12-25 11:53:12Z

1

You need an array initialization: fourier=[]

for i in range(1,125):
  fourier.append(sum(cos(2*pi*i*t/2001)*Trace))

answered Dec 25, 2014 at 11:53

nitishagar

9,4313 gold badges32 silver badges41 bronze badges

Comments

Dunes · Accepted Answer · 2014-12-25 12:33:52Z

There are several ways to build up a list of objects/numbers.

The simplest is a for loop:

# create a list of squared numbers
squares = []
for item in range(10):
    squares.append(item*item)

This can also be done using a "list comprehension":

# produces identical list to the above for loop
squares = [(item * item) for item in range(10)]

Finally, you are using numpy, this allows to do "vector" operations on arrays. The following is a simple example of the above square code.

eg.

numbers = numpy.array(range(10))
# or you could write -- numbers = numpy.arange(10)
squares = numbers * numbers

However, you can also do very complicated vector arithmetic as well. This allows you to create your Fourier transform very easily.

indices = array(range(125), ndmin=2).T
arr_fourier =  (cos(2*pi*indices*t/2001)*Trace).sum(axis=1)

Note that i has been replaced by indices. When multiplying the indices by t we create a transpose of the indices array so that we end up creating a 2d array as a result. At the end of the process you sum over the one of the axes (in this case 1) to reduce the 2d array back down to a 1d array of length 125.

Hackaholic · Accepted Answer · 2015-06-09 11:41:58Z

0

from pylab import *
from numpy import *
x = linspace(-5,5,100)
t = linspace(0,2000,2001)
w1=cos(2*pi*3*t/2001)
w2=cos(2*pi*10*t/2001)
fourier=[]
Trace = w1+w2
    for i in range(1,126):
        fourier.append(sum(cos(2*pi*i*t/2001)*Trace))
print fourier
f=linspace(1,125,125)
plot(f,fourier)
show()

edited Jun 9, 2015 at 11:41

Hackaholic

19.8k6 gold badges59 silver badges77 bronze badges

answered Dec 25, 2014 at 12:37

hariK

3,10915 silver badges19 bronze badges

Collectives™ on Stack Overflow

assign the outputs in a loop to a single list or array

4 Answers 4

5 Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

5 Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related