Matplotlib Plot Array As Function