Add try: except to create legend instead of colorbar (#1028)

Add try: except to create legend instead of colorbar (#1028) #2211