空 dask 数据帧的 len 引发异常

Question

我需要知道 dask 数据帧的长度，但如果我正在读取空文件 - 代码会产生异常：import dask.dataframe as ddif __name__ == '__main__':    ddf = dd.read_csv(r'empty_file.csv', names=['x'])    print(len(ddf))结果我收到错误：TypeError: 'numpy.float64' object cannot be interpreted as an integer但是，如果文件不为空，上面的代码可以正常工作。有没有更好的方法来获取 dask 数据帧的长度来处理文件为空的情况？更新：堆栈跟踪

Answer 1

您可以使用.empty来验证您的数据框是否为空。像这样：

import dask.dataframe as dd

if __name__ == '__main__':

    ddf = dd.read_csv(r'empty_file.csv', names=['x'])

    if not ddf.empty:

        print(len(ddf))

    else:

        print("Empty dataframe")