python - 如何在执行连接之前将染色体名称转换为 pyranges 中的相同格式

Question

我有多个 .bed 文件，我想对它们执行连接、交集等操作。我正在使用 pyranges库来读取 .bed 文件并执行这些操作。由于 .bed 文件允许命名带有或不带有“chr”前缀的染色体，我想在执行操作之前将不同 .bed 文件中的所有染色体名称格式化为相同的格式。因此，操作会产生预期的输出。
我试过，

>>> import pandas as pd
>>> import pyranges as pr
>>> df1 = pd.DataFrame({"Chromosome": ["chr1", "chr2"], "Start": [100, 200],
...                    "End": [150, 201]})
>>> py1 = pr.PyRanges(df1)
>>> df2 = pd.DataFrame({"Chromosome": ["1", "2"], "Start": [1000, 2000],
...                    "End": [1500, 20010]})
>>> py2 = pr.PyRanges(df2)
>>> def modify_chrom_series(df):
...    df.Chromosome = df.Chromosome.apply(lambda val: val.replace("chr", ""))
...    return df
>>> def fix_chrom(regions):
...    return regions.apply(modify_chrom_series)
>>> py1 = fix_chrom(py1)
>>> py1
+--------------+-----------+-----------+
|   Chromosome |     Start |       End |
|   (category) |   (int32) |   (int32) |
|--------------+-----------+-----------|
|            1 |       100 |       150 |
|            2 |       200 |       201 |
+--------------+-----------+-----------+
>>> py2 = fix_chrom(py2)
>>> py2

+--------------+-----------+-----------+
|   Chromosome |     Start |       End |
|   (category) |   (int32) |   (int32) |
|--------------+-----------+-----------|
|            1 |      1000 |      1500 |
|            2 |      2000 |     20010 |
+--------------+-----------+-----------+

>>> py1["1"]    
Empty PyRanges
>>> py1["chr1"]
+--------------+-----------+-----------+
|   Chromosome |     Start |       End |
|   (category) |   (int32) |   (int32) |
|--------------+-----------+-----------|
|            1 |       100 |       150 |
+--------------+-----------+-----------+

>>> py1.join(py2)
Empty PyRanges

使用上面的代码，染色体名称被格式化，但染色体名称在pyranges中的映射保持不变。因此，像 join 或 query py1["1"] 这样的操作不能按预期工作。
有没有办法使用 pyranges 获得所需的行为？

Answer 1

PyRanges中的数据类存储在多个位置。除了 .Chromosome , 你有 .dfs这是一个 dict .此 keys在此 dict当您执行 py1["1"] 时使用称呼。
您还需要更新字典

>>> df1 = pd.DataFrame({"Chromosome": ["chr1", "chr2"], "Start": [100, 200],
                       "End": [150, 201]})
>>> py1 = pr.PyRanges(df1)
>>> py1.dfs["1"] = py1.dfs['chr1']
>>> del py1.dfs['chr1']
>>> py1["1"]

+--------------+-----------+-----------+
| Chromosome   |     Start |       End |
| (category)   |   (int32) |   (int32) |
|--------------+-----------+-----------|
| chr1         |       100 |       150 |
+--------------+-----------+-----------+
Unstranded PyRanges object has 1 rows and 3 columns from 1 chromosomes.
For printing, the PyRanges was sorted on Chromosome.

请注意 name表中的染色体没有变化 - 这是因为，如上所述，数据存储在多个位置。
老实说 - 我不明白 PyRanges深深地，我不知道像这样更新数据是否安全。
我强烈建议到 预处理 您的数据仍然在 .bed格式。这将确保将数据正确导入到 pyranges。
编辑 1/8/20 : 答案基于 pre-bugfix行为，将来可能不需要。