Как получить подсписок списка между двумя словами списка в Python
Извините за мой английский. Я итальянец Начиная с такого списка:
words = ['tree', 'water', 'dog', 'soap', 'bike', 'cat', 'bird']
я хочу получить (в Python) подсписок между двумя словами, которые я ввожу с клавиатуры или в коде. Например, если я добавлю слова "вода" и "велосипед", я хочу получить подсписок:
words = ['water', 'dog', 'soap', 'bike']
или если список
words = ['tree', 'water', 'dog', 'soap', 'tree', 'cat', 'bird']
и я поставил слова "дерево" и "дерево", я хочу получить этот подсписок:
words = ['tree', 'water', 'dog', 'soap', 'tree']
Я также написал такую программу на C, но сейчас я не очень хорошо разбираюсь в Python. Это моя C версия.
Я надеюсь, что вы можете помочь мне, спасибо!
struct node {
char *key;
struct node *next;
struct node *prev;
};
typedef struct node node;
node *GetSublist(node *, char *, char *);
node *MakeStringList();
void PrintStringList(node *a);
node *GetSublist(node *a, char *k1, char *k2) {
node *p = a;
node *n1, *n2;
while (strcmp(p->key, k1) != 0) {
p = p->next;
if (p == NULL) {
return a;
}
}
n1 = p;
n2 = p->next;
if (n1->prev != NULL) {
while (a != n1) {
free(a);
a = a->next;
}
}
a->prev = NULL;
while (strcmp(n2->key, k2) != 0) {
n2 = n2->next;
if (n2 == NULL) {
return a;
}
}
if (n2->next != NULL) {
while (n2->next == NULL) {
free(n2->next);
n2 = n2->next;
}
n2->next = NULL;
}
return a;
}
int main(){
char *k1 = "dog";
char *k2 = "ball";
node *list1 = NULL;
list1 = MakeStringList();
PrintStringList(list1);
list1 = GetSublist(list1, k1, k2);
PrintStringList(list1);
return 0;
}
node *MakeStringList() {
node *a = NULL, *punt, *p;
int i;
int dim;
printf("Number of elements: ");
scanf("%d", &dim);
for (i=0; i<dim; i=i+1) {
punt = malloc( sizeof(node) );
punt->key = (char*)malloc(30*sizeof(char));
scanf( "%s", punt->key );
punt->next = NULL;
punt->prev = NULL;
if(a == NULL) {
a = punt;
p = punt;
} else {
p->next = punt;
punt->prev = p;
p = punt;
}
}
return a;
}
void PrintStringList(node *a) {
node *p = a;
printf("\nThe list is: { ");
while( p != NULL ) {
if (p->next == NULL)
printf("%s ", p->key);
else
printf("%s, ", p->key);
p = p->next;
}
printf("}\n\n");
}
1 ответ
Это может быть достигнуто с .index()
метод списков, и с обозначением среза..
words = ['tree', 'water', 'dog', 'soap', 'cat', 'bird']
start_index = words.index(start_word)
end_index = words.index(end_word)
sublist = words[start_index:end_index+1]
def sublist_two_words(array, start_word, end_word):
result = []
for word in array:
if result or word == start_word:
result.push(word)
if word == end_word:
break
return result
Таким образом, даже если нет end_word
он получает весь оставшийся список. Если я правильно понял вашу задачу.