matplotlib.scatter 색상 인수는 내가하기 matplotlib를 사용하여 스크립트를 세우고 다음 파이썬을 작성했습니다 NumPy와 배열

Question

을 수락하지 :

import pynbody as pyn 
import numpy as np 
import matplotlib.pyplot as plt 
import glob 

s = pyn.load('./ballsV2.00001') 
sl = s.g[np.where((s.g['z'] < 0.005) & (s.g['z']>-0.005))] 

sx = s.s['x'][0] 
sy = s.s['y'][0] 
sz = s.s['z'][0] 
r2 = ((s.g['x']-sx)**2+(s.g['y']-sy)**2+(s.g['z']-sz)**2) 
Flux = np.array(1./(4*np.pi*r2)*np.exp(-1*7.00114988051*np.sqrt(r2))) 

print(type(np.log10(sl['radFlux']))) 
print(type(np.log10(Flux))) 

plt.figure(figsize = (15,12)) 
#plt.scatter(sl['x'],sl['y'],c=np.log10(sl['radFlux']),s=75,edgecolors='none', marker = '.',vmin=-6,vmax=1) 
plt.scatter(sl['x'],sl['y'],c=np.log10(Flux),s=75,edgecolors='none', marker = '.',vmin=-8,vmax=4) 
plt.xlim([-0.5,0.5]) 
plt.ylim([-0.5,0.5]) 
plt.xlabel("x") 
plt.ylabel("y") 
plt.colorbar(label="log(Code Flux)") 
plt.savefig('./ballsV2_0.1.pdf') 
plt.savefig('./ballsV2_0.1.png') 
plt.show() 
plt.close() 

나는 다음과 같은 오류 얻을 스크립트를 실행하면 :

foo@bar ~/Data/RadTransfer/Scaling_Tests/ballsV2 $ py 
balls.py 
balls.py:15: RuntimeWarning: divide by zero encountered in log10 
    print(type(np.log10(sl['radFlux']))) 
<class 'numpy.ndarray'> 
<class 'numpy.ndarray'> 

Traceback (most recent call last): 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/colors.py", line 141, in to_rgba 
    rgba = _colors_full_map.cache[c, alpha] 
KeyError: (-4.1574455411341349, None) 

During handling of the above exception, another exception occurred: 

Traceback (most recent call last): 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/colors.py", line 192, in _to_rgba_no_colorcycle 
    c = tuple(map(float, c)) 
TypeError: 'numpy.float64' object is not iterable 

During handling of the above exception, another exception occurred: 

Traceback (most recent call last): 
    File "balls.py", line 17, in <module> 
    plt.scatter(sl['x'],sl['y'],c=np.log10(Flux),s=75,edgecolors='none', marker = '.',vmin=-8,vmax=4) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/pyplot.py", line 3435, in scatter 
    edgecolors=edgecolors, data=data, **kwargs) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/__init__.py", line 1892, in inner 
    return func(ax, *args, **kwargs) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/axes/_axes.py", line 4028, in scatter 
    alpha=alpha 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/collections.py", line 890, in __init__ 
    Collection.__init__(self, **kwargs) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/collections.py", line 139, in __init__ 
    self.set_facecolor(facecolors) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/collections.py", line 674, in set_facecolor 
    self._set_facecolor(c) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/collections.py", line 659, in _set_facecolor 
    self._facecolors = mcolors.to_rgba_array(c, self._alpha) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/colors.py", line 237, in to_rgba_array 
    result[i] = to_rgba(cc, alpha) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/colors.py", line 143, in to_rgba 
    rgba = _to_rgba_no_colorcycle(c, alpha) 
    File "/home/grondjj/anaconda3/lib/python3.6/site-packages/matplotlib/colors.py", line 194, in _to_rgba_no_colorcycle 
    raise ValueError("Invalid RGBA argument: {!r}".format(orig_c)) 
ValueError: Invalid RGBA argument: -4.1574455411341349 

를 무시 제로 물건으로 나누면, 문제는 산점도 함수는 색상을 매핑하는 값의 배열을 사용하지 않습니다. 이상한 점은 위의 주석 된 산점도 명령이 정상적으로 실행된다는 것입니다. 유일한 차이점은 내가 전달하는 값 배열입니다. 나는 그들을 같은 타입으로 던지게했다 (그들은 모두 <class 'numpy.ndarray'>이다). 또한 값 자체는 Flux 배열에서 ~ 4000과 1E-7 사이의 범위에서 더 정숙합니다. 0으로 나누는 오류가 있고 그 중 하나가 작동하는 것은 np.log10(sl['radFlux']뿐입니다. 어떤 제안?

Answer 1

Flux 및 np.log10(sl['radFlux'])은 다른 길이가되었다. sl (슬라이스가 s)은 r2을 계산하는 데 사용되지 않으므로 Flux은 큰 것으로 끝납니다. matplotlib가 색상 배열이 scatter x 및 y 배열과 길이가 같고 x 및 y 배열이 다른 길이 일 때와 같은 오류 메시지가 있는지 검사하면 좋을 것입니다.