so im trying to use pandas instead of a for loop to count the number of movies in a given range of year. Assume by data frame has 2 columns and 'year' is the column name at column 2
I solved it using a for loop but how would i do it by using pandas only?
def movie_made(beginning, end): movie = pd.read_scv('title.csv') count = 0 for i in move['year']: if beginning <= i and end <=i: count = count + 1 return count
This allows me to count all the movies within a given year, but im wondering if there is a better way using the pandas infrastructure for reading from a data base.
You could do something like this:
import pandas as pd df = pd.DataFrame(data=list(range(1980, 2001)), columns=['year']) beginning, end = 1998, 2000 def movie_made(df, beginning, end): return len(df[(beginning <= df['year']) & (df['year'] <= end)].index) print(movie_made(df, beginning, end))