DataFrame用loc函数选择多列后组成的数据还是Dataframe,选取一列组成的数据是Series(备注:代码注释“#”后为运行结果)。
lat=sim_poi.loc[sim_poi.station == '上海火车站','gps_lat']
print(lat)
# 188 31.250757
# Name: gps_lat, dtype: float64
print("type:",type(lat))
# type: <class 'pandas.core.series.Series'>
取Series包含index和value,取其中的value:
lat_value=lat.values
print(type(lat_value))
# <class 'numpy.ndarray'>
print(lat_value)
#[ 31.25075675]
print(lat_value[0])
#31.2507567527
返回的是一维numpy数组。对于numpy数组判断是否为空,可以用数组中元素个数size是否为0:
if lat_value.size:
print(lat_value.size)
#1
另外,对于loc取出的DataFrame,取出其中的Values得到一个多维数组,: