将变量分配给 Neo4jRestClient 时出现 IndexError

IndexError when assigning variables to Neo4jRestClient

我正在尝试访问 SciKit Allele 中的基因组数据,这是一种基于 Numpy 的用于基因组数据的工具。

我不太擅长 python,但我正在尝试遍历每个变体并提取数组中的相关列,然后使用 Neo4j Rest Client 在 Neo4j 数据库中创建节点。

下面的代码生成一个包含所有变体和所有数据类型的数组:

variants = allel.VariantChunkedTable(callset[chrom]['variants'], names=['AC','AF_AFR', 'AF_AMR', 'AF_ASN', 'AF_EUR', 'AF_MAX', 'CGT', 'CLR', 'CSQ', 'DP', 'DP4', 'ESP_MAF', 'FILTER_LowQual', 'FILTER_MinHWE', 'FILTER_MinVQSLOD', 'FILTER_PASS', 'HWE', 'ICF', 'ID', 'IS', 'PC2', 'PCHI2', 'POS', 'PR', 'QCHI2', 'QUAL', 'REF', 'ALT', 'INDEL', 'SHAPEIT', 'SNP_ID', 'TYPE', 'UGT', 'VQSLOD', 'dbSNPmismatch', 'is_snp', 'numalt', 'svlen'], index='POS')

我(认为我)声明了数组形式的变量如下:

pos = variants['POS'][:]
alt = variants['ALT'][:]
dp = variants['DP'][:]
ac = variants['AC'][:]
type = variants['TYPE'][:]
svlen = variants['svlen'][:]
qual = variants['QUAL'][:]
vq = variants['VQSLOD'][:]

这些变量创建数组,如:

In: pos
Out: array([    28590,     50481,     52152, ..., 249225077, 249229702,
       249231222], dtype=int32)

我现在正在尝试访问每一行的变量,但似乎不知道如何操作。我目前的尝试看起来像这样(前 10 行):

for variant in variants[0:10]:
    a1 = db.nodes.create(pos=pos[variant], bp=alt[variant][0], DP=dp[variant], AC=ac[variant][0], type=type[variant][0], svlen=svlen[variant][0], qual=qual[variant], vqslod=vq[variant])
    a1.relationships.create("belongs_to", c1)

不幸的是,这会出现以下错误:

IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

谁能帮我弄清楚如何获取每个属性的特定变量?

由于没有可重现的示例我无法遵循您的代码,因此我不得不根据 scikit-allel 文档创建一个:

https://scikit-allel.readthedocs.io/en/stable/model/chunked.html#variantchunkedtable

import h5py
import allel
import os

#cleanup
h5_file = 'callset.h5'
os.remove(h5_file) if os.path.exists(h5_file) else None

chrom = [b'chr1', b'chr1', b'chr2', b'chr2', b'chr3', b'chr3']
pos = [2, 7, 3, 9, 6, 11]
dp = [35, 12, 78, 22, 99, 96]
qd = [4.5, 6.7, 1.2, 4.4, 2.8, 3.2]
ac = [(1, 2), (3, 4), (5, 6), (7, 8), (9, 10), (11, 12)]

with h5py.File(h5_file, mode='w') as h5f:
    h5g = h5f.create_group('/3L/variants')
    h5g.create_dataset('CHROM', data=chrom, chunks=True)
    h5g.create_dataset('POS', data=pos, chunks=True)
    h5g.create_dataset('DP', data=dp, chunks=True)
    h5g.create_dataset('QD', data=qd, chunks=True)
    h5g.create_dataset('AC', data=ac, chunks=True)


callset = h5py.File(h5_file, mode='r')
variants = allel.VariantChunkedTable(callset['/3L/variants'],
                                names=['CHROM', 'POS', 'AC', 'QD', 'DP'])

因此,variants 变量(在此示例中只有 6 行)看起来像这样:

>>> variants
<VariantChunkedTable shape=(6,) dtype=[('CHROM', 'S4'), ('POS', '<i8'), ('AC', '<i8', (2,)), ('QD', '<f8'), ('DP', '<i8')] 
nbytes=264 cbytes=264 cratio=1.0 values=h5py._hl.group.Group>
    CHROM   POS AC      QD  DP
0   b'chr1' 2   [1 2]   4.5 35
1   b'chr1' 7   [3 4]   6.7 12
2   b'chr2' 3   [5 6]   1.2 78
3   b'chr2' 9   [7 8]   4.4 22
4   b'chr3' 6   [ 9 10] 2.8 99
5   b'chr3' 11  [11 12] 3.2 96

您已正确定义 posaltdp 等数组(即 pos = variants['POS'][:] 等)

然后,在你的循环中,我假设你的目标是遍历 variants 变量的前 10 行,从每一行中获取一些值,例如pos[variant]ac[variant][0]dp[variant] 并使用这些属性在 GraphDatabase 中创建一个新节点。

按照您目前编写循环的方式,每次迭代都会从 variants 中获取整行,并尝试将其用作索引以访问 pos 中的元素,alt, ... 数组,抛出错误。

正确的做法是遍历数字索引;在我的示例中,要遍历 variants 变量的所有 6 行,您应该 运行:

for i in range(len(variants)):
    print(f"> Row {i}")
    print(pos[i])
    print(dp[i])
    print(ac[i][0])

然后可以将 pos[i]、dp[i] 等值以 name=value 对的形式馈送到 db.nodes.create。 当然,对于前 10 行,您只需要使用 for i in range(10).