U f{j@s0ddlZddlmZmZddlmZddZdS)N)_validate_type _check_option)int_likecsddlmt|}t|D]\}}d|d}t|dtjjf|t|t rxj ddgdt |t |ft d }n^t|d |jd g|jd|jdkrt|d |jt|jsΈ|}n|}d |j|j|jk<||jdkstd |||<qtfdd|Ds.ttdd|Dt tfddtt|Dt }tdd|Dt |}|}t}t| }t!d ||ft } t"|t#} t!||t } d} t|D]\}}t$dgt} t$dgt}|j| |<|j||<|jj|jjkrB|jjksHntt$| | ||}|t%| &|t%|&g| dd|f<t'|j||| |<| ||7} | |(rtd| |<q| st|&| dd|df<d| |d<| | f|j)|j)f}|S)akCreate a sparse binary adjacency/neighbors matrix. Parameters ---------- *structure : list The adjacency along each dimension. Each entry can be: - ndarray or scipy.sparse.spmatrix A square binary adjacency matrix for the given dimension. For example created by :func:`mne.channels.find_ch_adjacency`. - int The number of elements along the given dimension. A lattice adjacency will be generated, which is a binary matrix reflecting that element N of an array is adjacent to elements at indices N - 1 and N + 1. Returns ------- adjacency : scipy.sparse.coo_matrix, shape (n_features, n_features) The square adjacency matrix, where the shape ``n_features`` corresponds to the product of the length of all dimensions. For example ``len(times) * len(freqs) * len(chans)``. See Also -------- mne.channels.find_ch_adjacency mne.channels.read_ch_adjacency Notes ----- For 4-dimensional data with shape ``(n_obs, n_times, n_freqs, n_chans)``, you can specify **no** connections among elements in a particular dimension by passing a matrix of zeros. For example: >>> import numpy as np >>> from scipy.sparse import diags >>> from mne.stats import combine_adjacency >>> n_times, n_freqs, n_chans = (50, 7, 16) >>> chan_adj = diags([1., 1.], offsets=(-1, 1), shape=(n_chans, n_chans)) >>> combine_adjacency( ... n_times, # regular lattice adjacency for times ... np.zeros((n_freqs, n_freqs)), # no adjacency between freq. bins ... chan_adj, # custom matrix, or use mne.channels.find_ch_adjacency ... ) # doctest: +NORMALIZE_WHITESPACE <5600x5600 sparse matrix of type '' with 27076 stored elements in COOrdinate format> rsparsez structure[]zint-like)r )offsetsshapeZdtypez.ndimrz must be square, got shape gz#All adjacency values must be 0 or 1c3s|]}t|jVqdS)N) isinstance coo_matrix.0dimrS/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/stats/_adjacency.py Xsz$combine_adjacency..cSsg|]}|jdqS)rr )rdrrr Ysz%combine_adjacency..c s4g|],}ttd||ddgqS)Nr )npprodZ concatenate)rdirrrrZscSsg|] }|jjqSr)datasizerrrrr\sNTg?)*Zscipyrlist enumeraterrZndarrayZspmatrixr rZdiagsintfloatZtocoorndimr ValueErrorrcopyrrowcolZeliminate_zerosallAssertionErrorarrayrangelensumrZarangeZreshapeemptyZzerosboolslicetupleZravelZtileanyr)Z structurerrnameZn_othersZn_eachZn_offZn_diagZverticesedgesusedweightsoffsetZs_lZs_rslgraphr)r rrcombine_adjacencysx0             *     r8)numpyrutilsrrZ utils.checkrr8rrrrs