Более быстрая генерация PyRanges в Python

У меня есть список начальных координат на одной хромосоме для объектов фиксированного размера, и я пытаюсь создать объект PyRanges.

Я рассчитал время генерации объекта PyRanges по списку из 125 координат, и это заняло около 3,5 мс. Это оказалось медленнее, чем ожидалось (я впервые использую эту библиотеку), поэтому я попытался измерить скорость одного и того же процесса в списке разного размера.

Вот результаты тестов производительности:

N = 1: 3.03ms
N = 10: 2.96ms
N = 100: 3.33ms
N = 125: 3.24ms
N = 200: 3.11ms
N = 500: 3.12ms
N = 10000: 6.86ms
N = 100000: 32.6ms

Похоже, что для создания объекта PyRanges требуется базовое время (при N = 1 это все еще занимает некоторое время), а затем, хотя время, кажется, зависит от количества функций, отношение не кажется слишком резкий. Действительно, создание объекта PyRanges из 10000 элементов занимает всего в 2 раза больше времени, чем создание объекта всего из 10.

Это код, который я использую:

chr = "chrX"
size = 10
N = 1
points = np.array([random.randint(0, 1000000) for i in range(N)])

genomic_range = pr.PyRanges(
                chromosomes= chr,
                starts = points,
                ends = points + size - 1
            )

Я делаю что-то неправильно? Почему создание PyRange занимает это время даже для нескольких элементов?

Можете ли вы добавить минимальный пример воспроизводящий для своего теста, чтобы мы могли легко попытаться воспроизвести проблему на нашей машине?

Jérôme Richard 18.03.2022 12:43

Я добавил код.

Fabrizio 18.03.2022 14:06
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
1
2
29
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Короче говоря, PyRanges использует Pandas внутри, что очень медленно для небольших входных данных..


Вызов функции идет в _init, который вызывает create_pyranges_df, который сам выполняет следующую строку:

chromosomes = pd.Series([chromosomes] * len(starts), dtype = "category")

Эта строка занимает на моей машине 0,25 мс, что очень медленно для такого очень маленького ввода (я ожидаю, что это будет как минимум в 100 раз быстрее). dtype = "category", кажется, причина, почему он медленный.

Самая медленная часть кода находится здесь:

for s in columns:
    if isinstance(s, pd.Series):
        s = pd.Series(s.values, index=idx)  # This line is executed several time
    else:
        s = pd.Series(s, index=idx)

    series_to_concat.append(s)

Наличие 10 столбцов и ~ 0,2 мс для создания каждого объекта pd.Series приводит к тому, что ~ 2 мс тратится почти на ничегонеделание...

Другие вопросы по теме