How to list the csv files in a folder using Python

#get csv list in a folder

import os
n = 1
arr = os.listdir()
print(" > CSV files in your folder:")
for i in arr:
    if i[-4:]==".csv":
        print (" (",n,")",i)
        n = n+1