Test Pandas ,numpy csv

#Antonio Villanueva Segura test de pandas ,numpy et csv
# -*- coding: utf-8

import pandas as pd
import numpy
import csv

def pago (piezas,precio):
  """calcula el precio segun la cantidad"""
  return piezas*precio


"""Crea un array de datos, producto y precios por columnas"""
"""Primera columna producto"""
prod=numpy.array( ['peras','manzanas','melocotones','fresas','cerezas'] )
"""Segunda columna precios """
prec=numpy.array([5,7,2,10,9])

"""creo el array con las columnas anteriores """
datos=numpy.column_stack((prod,prec))

"""Crea  indices y columnas + datos"""
df=pd.DataFrame(datos,columns = ['producto','precio'])

"""Escribe datos a un fichero externo *csv"""
#df.to_csv('precios.csv',index=False)#Elimino indice

"""
producto=input("Introduce tu compra :")
piezas=int(input("Introduce la cantidad :"))
"""

""" Test directo de compra de 6 fresas a 10 euros """
producto='fresas'
piezas=6

"""precio """
precio=int (df[ df['producto']==producto ]['precio']) #Elimina indice coge precio

print (pago (piezas,precio))